Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <gemm_operand.h>
Public Types | |
typedef ThreadBlockTile_ | ThreadBlockTile |
Shape of GEMM thread block tile (K, N, M) More... | |
typedef platform::conditional< kKstrided, Shape< 1, ThreadBlockTile::kD, GetExtent< Usage, ThreadBlockTile >::kExtent >, Shape< 1, GetExtent< Usage, ThreadBlockTile >::kExtent, ThreadBlockTile::kD > >::type | Shape |
Map the ThreadBlockShape onto (kH, kW) dimensions for A and B operand. More... | |
Static Public Attributes | |
static GemmOperand::Kind const | kUsage = Usage |
Identifies multiplicand. More... | |
static MatrixLayout::Kind const | kLayout = Layout |
Layout of tile. More... | |
static bool const | kKstrided = (kUsage == GemmOperand::kA ^ kLayout == MatrixLayout::kRowMajor) |
Determines the shape of a multiplicand tile in terms of strided (H) and contiguous (W) dimensions
typedef platform::conditional< kKstrided, Shape<1, ThreadBlockTile::kD, GetExtent<Usage, ThreadBlockTile>::kExtent>, Shape<1, GetExtent<Usage, ThreadBlockTile>::kExtent, ThreadBlockTile::kD> >::type cutlass::gemm::GemmMultiplicandTraits< ThreadBlockTile_, Usage, Layout >::Shape |
typedef ThreadBlockTile_ cutlass::gemm::GemmMultiplicandTraits< ThreadBlockTile_, Usage, Layout >::ThreadBlockTile |
|
static |
|
static |
|
static |