Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Class for storing a tile in memory and accessing it through a tensor ref.
#include <tile_allocation.h>
Public Types | |
typedef Scalar_ | Scalar |
Scalar element. More... | |
typedef StorageType< sizeof(Scalar)>::Type | Storage |
The actual storage (may differ from the scalar type) More... | |
typedef Shape_ | Shape |
Size of the allocation in units of scalars. More... | |
typedef ShapeStrides< Shape, 1 >::Shape | Strides |
Strides. More... | |
typedef TensorRef< Scalar const, 4 > | ConstTensorRef |
Defines the tensor reference for this allocation. More... | |
typedef TensorRef< Scalar, 4 > | TensorRef |
Defines the tensor reference for this allocation. More... | |
typedef TensorView< Scalar const, 4 > | ConstTensorView |
View of memory. More... | |
typedef TensorView< Scalar, 4 > | TensorView |
View of memory. More... | |
Public Member Functions | |
CUTLASS_DEVICE Scalar * | data () |
Returns a pointer to the raw data. More... | |
CUTLASS_DEVICE Scalar const * | data () const |
Returns a const pointer to the raw data. More... | |
CUTLASS_DEVICE TensorRef | reference () |
Returns a TensorRef object pointing to the data. More... | |
CUTLASS_DEVICE ConstTensorRef | reference () const |
Returns a TensorRef object pointing to the data. More... | |
CUTLASS_DEVICE TensorView | view () |
Returns a TensorView object pointing to the data. More... | |
CUTLASS_DEVICE ConstTensorView | view () const |
Returns a TensorView object pointing to the data. More... | |
Public Attributes | |
Storage | storage [Shape::kD][Shape::kH][Shape::kW][Shape::kC] |
Storage. More... | |
typedef TensorRef<Scalar const, 4> cutlass::TileAllocation< Scalar_, Shape_ >::ConstTensorRef |
typedef TensorView<Scalar const, 4> cutlass::TileAllocation< Scalar_, Shape_ >::ConstTensorView |
typedef Scalar_ cutlass::TileAllocation< Scalar_, Shape_ >::Scalar |
typedef Shape_ cutlass::TileAllocation< Scalar_, Shape_ >::Shape |
typedef StorageType<sizeof(Scalar)>::Type cutlass::TileAllocation< Scalar_, Shape_ >::Storage |
typedef ShapeStrides<Shape, 1>::Shape cutlass::TileAllocation< Scalar_, Shape_ >::Strides |
typedef TensorRef<Scalar, 4> cutlass::TileAllocation< Scalar_, Shape_ >::TensorRef |
typedef TensorView<Scalar, 4> cutlass::TileAllocation< Scalar_, Shape_ >::TensorView |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
Storage cutlass::TileAllocation< Scalar_, Shape_ >::storage[Shape::kD][Shape::kH][Shape::kW][Shape::kC] |