int64_t LongIndex
Long index type used for offsets. 
Definition: vector.h:49
static int const kStrideRank
Rank of stride vector. 
Definition: vector.h:43
Definition: aligned_buffer.h:35
A Coord is a coordinate of arbitrary rank into a tensor or matrix. 
CUTLASS_HOST_DEVICE Coord< 1 > make_Coord(int _0)
Helper to make a 2-element coordinate. 
Definition: coord.h:387
CUTLASS_HOST_DEVICE PackedVectorLayout()
Definition: vector.h:70
CUTLASS_HOST_DEVICE Stride stride() const 
Returns the stride of the layout. 
Definition: vector.h:86
static int const kRank
Logical rank of tensor. 
Definition: vector.h:40
int32_t Index
Index type used for coordinates. 
Definition: vector.h:46
static CUTLASS_HOST_DEVICE PackedVectorLayout packed(TensorCoord const &size)
Helper returns a layout to a tightly packed tensor. 
Definition: vector.h:74
#define CUTLASS_HOST_DEVICE
Definition: cutlass.h:89
CUTLASS_HOST_DEVICE LongIndex operator()(TensorCoord const &coord) const 
Returns the offset of a coordinate in linear memory. 
Definition: vector.h:80
Statically-sized array specifying Coords within a tensor. 
Definition: coord.h:43
Tensor layout for densely packed vectors. 
Definition: vector.h:37
Basic include for CUTLASS. 
CUTLASS_HOST_DEVICE LongIndex capacity(TensorCoord const &size) const 
Compute the number of contiguous elements needed to store a tensor with the given size...
Definition: vector.h:92