Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <gemm_shared_tile.h>
Classes | |
struct | ThreadOffset |
Public Types | |
typedef nv_std::remove_const< Scalar_ >::type | Scalar |
The scalar. More... | |
typedef Scalar_ * | Pointer |
The pointer. More... | |
typedef ReshapeTile< Tile_, kScalarsPerSts_ >::Tile | Tile |
The tile. More... | |
typedef Threads_ | Threads |
The threads. More... | |
typedef Shape< 0, ShapeCount< Tile >::kWc, Tile::kC, kScalarsPerSts_ > | ThreadsStrides |
The strides to compute the base position of the thread. More... | |
typedef Shape< 1, Tile::kH/Threads::kH, Tile::kW/Threads::kW, Tile::kC/Threads::kC/kScalarsPerSts > | Iterations |
The number of iterations needed to load/store the tile. More... | |
typedef Shape< 0, Threads::kH *ShapeCount< Tile >::kWc, Threads::kW *kScalarsPerSts > | Delta |
The strides in each dimension between different loads/stores. More... | |
Static Public Attributes | |
static int const | kSkew = 0 |
The skew. More... | |
static int const | kScalarsPerSts = kScalarsPerSts_ |
The number of scalars per LDG/STG. More... | |
static MemorySpace::Kind const | kMemorySpace = MemorySpace::kShared |
The memory space. More... | |
typedef Shape<0, Threads::kH * ShapeCount<Tile>::kWc, Threads::kW * kScalarsPerSts> cutlass::gemm::GemmSharedStoreIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_ >::Delta |
typedef Shape<1, Tile::kH / Threads::kH, Tile::kW / Threads::kW, Tile::kC / Threads::kC / kScalarsPerSts> cutlass::gemm::GemmSharedStoreIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_ >::Iterations |
typedef Scalar_* cutlass::gemm::GemmSharedStoreIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_ >::Pointer |
typedef nv_std::remove_const<Scalar_>::type cutlass::gemm::GemmSharedStoreIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_ >::Scalar |
typedef Threads_ cutlass::gemm::GemmSharedStoreIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_ >::Threads |
typedef Shape<0, ShapeCount<Tile>::kWc, Tile::kC, kScalarsPerSts_> cutlass::gemm::GemmSharedStoreIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_ >::ThreadsStrides |
typedef ReshapeTile<Tile_, kScalarsPerSts_>::Tile cutlass::gemm::GemmSharedStoreIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_ >::Tile |
|
static |
|
static |
|
static |