Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Manages a pair of tile allocations as if they are one allocation.
#include <tile_allocation.h>
Public Types | |
typedef First_ | First |
First tensor allocation. More... | |
typedef Second_ | Second |
Second tensor allocation. More... | |
typedef ZipTensorRef< typename First::TensorRef, typename Second::TensorRef > | TensorRef |
Defines the tensor reference for this allocation. More... | |
typedef ZipTensorRef< typename First::ConstTensorRef, typename Second::ConstTensorRef > | ConstTensorRef |
Defines the tensor reference for this allocation. More... | |
Public Member Functions | |
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... | |
Public Attributes | |
First | first |
First tensor allocation. More... | |
Second | second |
Second tensor allocation. More... | |
typedef ZipTensorRef<typename First::ConstTensorRef, typename Second::ConstTensorRef> cutlass::ZipTileAllocation< First_, Second_ >::ConstTensorRef |
typedef First_ cutlass::ZipTileAllocation< First_, Second_ >::First |
typedef Second_ cutlass::ZipTileAllocation< First_, Second_ >::Second |
typedef ZipTensorRef<typename First::TensorRef, typename Second::TensorRef> cutlass::ZipTileAllocation< First_, Second_ >::TensorRef |
|
inline |
|
inline |
First cutlass::ZipTileAllocation< First_, Second_ >::first |
Second cutlass::ZipTileAllocation< First_, Second_ >::second |