41 template <
typename Scalar_,
typename Shape_>
107 make_Coord(Strides::kD, Strides::kH, Strides::kW, Strides::kC),
116 make_Coord(Strides::kD, Strides::kH, Strides::kW, Strides::kC),
124 template <
typename First_,
typename Second_>
CUTLASS_DEVICE TensorRef reference()
Returns a TensorRef object pointing to the data.
Definition: tile_allocation.h:159
ZipTensorRef< typename First::ConstTensorRef, typename Second::ConstTensorRef > ConstTensorRef
Defines the tensor reference for this allocation.
Definition: tile_allocation.h:141
CUTLASS_DEVICE TensorView view()
Returns a TensorView object pointing to the data.
Definition: tile_allocation.h:104
Shape< Shape_::kH *Shape_::kW *Shape_::kC, Shape_::kW *Shape_::kC, Shape_::kC, elementsPerAccess > Shape
Definition: shape.h:170
Defines a structure containing strides, bounds, and a pointer to tensor data.
Definition: zip_tensor_ref.h:38
TensorView< Scalar, 4 > TensorView
View of memory.
Definition: tile_allocation.h:69
TensorRef< Scalar, 4 > TensorRef
Defines the tensor reference for this allocation.
Definition: tile_allocation.h:63
CUTLASS_DEVICE Scalar const * data() const
Returns a const pointer to the raw data.
Definition: tile_allocation.h:88
CUTLASS_HOST_DEVICE Coord< 1 > make_Coord(int _0)
Helper to make a 2-element coordinate.
Definition: coord.h:368
CUTLASS_DEVICE TensorRef reference()
Returns a TensorRef object pointing to the data.
Definition: tile_allocation.h:92
Second_ Second
Second tensor allocation.
Definition: tile_allocation.h:134
Storage storage[Shape::kD][Shape::kH][Shape::kW][Shape::kC]
Storage.
Definition: tile_allocation.h:76
CUTLASS_DEVICE ConstTensorView view() const
Returns a TensorView object pointing to the data.
Definition: tile_allocation.h:113
ZipTensorRef< typename First::TensorRef, typename Second::TensorRef > TensorRef
Defines the tensor reference for this allocation.
Definition: tile_allocation.h:137
ShapeStrides< Shape, 1 >::Shape Strides
Strides.
Definition: tile_allocation.h:57
Defines a structure containing strides and a pointer to tensor data.
static int const kH
The height of the cube.
Definition: shape.h:68
Second second
Second tensor allocation.
Definition: tile_allocation.h:151
Definition: fragment.h:76
static int const kC
The number of scalars per element.
Definition: shape.h:72
TensorRef< Scalar const, 4 > ConstTensorRef
Defines the tensor reference for this allocation.
Definition: tile_allocation.h:60
First first
First tensor allocation.
Definition: tile_allocation.h:148
CUTLASS_DEVICE ConstTensorRef reference() const
Returns a TensorRef object pointing to the data.
Definition: tile_allocation.h:98
Defines a structure containing a pair of TensorRef-like objects.
Scalar_ Scalar
Scalar element.
Definition: tile_allocation.h:48
Manages a pair of tile allocations as if they are one allocation.
Definition: tile_allocation.h:125
Shape_ Shape
Size of the allocation in units of scalars.
Definition: tile_allocation.h:54
Class for storing a tile in memory and accessing it through a tensor ref.
Definition: tile_allocation.h:42
CUTLASS_DEVICE Scalar * data()
Returns a pointer to the raw data.
Definition: tile_allocation.h:84
TensorView< Scalar const, 4 > ConstTensorView
View of memory.
Definition: tile_allocation.h:66
CUTLASS_DEVICE ConstTensorRef reference() const
Returns a TensorRef object pointing to the data.
Definition: tile_allocation.h:163
static int const kW
The width of the cube.
Definition: shape.h:70
static int const kD
The depth of the cube.
Definition: shape.h:66
Defines Shape implementing the Layout concept for representing a 4D hypercube of objects.
StorageType< sizeof(Scalar)>::Type Storage
The actual storage (may differ from the scalar type)
Definition: tile_allocation.h:51
Defines Fragment, a statically-sized array for storing parts of matrices within a thread's registers...
First_ First
First tensor allocation.
Definition: tile_allocation.h:131