|
Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <gemm_traits.h>
Public Types | |
| typedef GemmConfig_::ScalarB | Scalar |
| The input scalar. More... | |
| typedef GemmConfig_::MultiplyAdd::ScalarB | MultiplyAddScalar |
| The scalar stored in shared memory. More... | |
| typedef GemmGlobalTileTraits< GemmOperand::kB, MatrixLayout::kColumnMajor, Scalar const, Shape< 1, GemmConfig_::OutputTile::kH, GemmConfig_::OutputTile::kD >, Shape< 1, GemmConfig_::kThreads/GemmConfig_::OutputTile::kD, GemmConfig_::OutputTile::kD >, GemmConfig_::kScalarsPerLdgB > | GlobalTileTraits |
| The traits class to build the iterator to load data from global memory for B^N. More... | |
Static Public Attributes | |
| static MatrixLayout::Kind const | kLayout = MatrixLayout::kColumnMajor |
| The layout. More... | |
| static int const | kScalarsIn4B = sizeof(MultiplyAddScalar) > 4 ? 1 : 4 / sizeof(MultiplyAddScalar) |
| The number of scalars in 4B. More... | |
| static int const | kSkewB |
| The skew for B. More... | |
| typedef GemmGlobalTileTraits< GemmOperand::kB, MatrixLayout::kColumnMajor, Scalar const, Shape<1, GemmConfig_::OutputTile::kH, GemmConfig_::OutputTile::kD>, Shape<1, GemmConfig_::kThreads / GemmConfig_::OutputTile::kD, GemmConfig_::OutputTile::kD>, GemmConfig_::kScalarsPerLdgB> cutlass::gemm::GemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >::GlobalTileTraits |
| typedef GemmConfig_::MultiplyAdd::ScalarB cutlass::gemm::GemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >::MultiplyAddScalar |
| typedef GemmConfig_::ScalarB cutlass::gemm::GemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >::Scalar |
|
static |
|
static |
|
static |
1.8.14