|
typedef GemmTraits< GemmConfig_, SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA< kLayoutA_, GemmConfig_ >, GemmTileTraitsHelperB< kLayoutB_, GemmConfig_ >, Index_ > ::GlobalLoadStreamA, SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA< kLayoutA_, GemmConfig_ >, GemmTileTraitsHelperB< kLayoutB_, GemmConfig_ >, Index_ > ::GlobalLoadStreamB, SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA< kLayoutA_, GemmConfig_ >, GemmTileTraitsHelperB< kLayoutB_, GemmConfig_ >, Index_ > ::SharedLoadStreamA, SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA< kLayoutA_, GemmConfig_ >, GemmTileTraitsHelperB< kLayoutB_, GemmConfig_ >, Index_ > ::SharedLoadStreamB, GemmEpilogue< GemmEpilogueTraits_ >, IdentityBlockSwizzle, Index_, ClearAccumulators< GemmConfig_::Accumulators::Element > > | This_ |
| This traits. More...
|
|
typedef cutlass::gemm::Gemm< This_ > | KernelClass |
| The struct that consumes this Traits. More...
|
|
typedef GemmConfig_ | GemmConfig |
| The configuration. More...
|
|
typedef GemmConfig::OutputTile | OutputTile |
| The output tile. More...
|
|
typedef SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA< kLayoutA_, GemmConfig_ >, GemmTileTraitsHelperB< kLayoutB_, GemmConfig_ >, Index_ > ::GlobalLoadStreamA | GlobalLoadStreamA |
| The stream to load A from global memory to shared memory. More...
|
|
typedef SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA< kLayoutA_, GemmConfig_ >, GemmTileTraitsHelperB< kLayoutB_, GemmConfig_ >, Index_ > ::GlobalLoadStreamA ::Scalar | ScalarA |
| The scalar for A. More...
|
|
typedef SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA< kLayoutA_, GemmConfig_ >, GemmTileTraitsHelperB< kLayoutB_, GemmConfig_ >, Index_ > ::GlobalLoadStreamB | GlobalLoadStreamB |
| The stream to load B from global memory to shared memory. More...
|
|
typedef SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA< kLayoutA_, GemmConfig_ >, GemmTileTraitsHelperB< kLayoutB_, GemmConfig_ >, Index_ > ::GlobalLoadStreamB ::Scalar | ScalarB |
| The scalar for B. More...
|
|
typedef SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA< kLayoutA_, GemmConfig_ >, GemmTileTraitsHelperB< kLayoutB_, GemmConfig_ >, Index_ > ::SharedLoadStreamA | SharedLoadStreamA |
| The iterator for A to load from shared memory. More...
|
|
typedef SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA< kLayoutA_, GemmConfig_ >, GemmTileTraitsHelperB< kLayoutB_, GemmConfig_ >, Index_ > ::SharedLoadStreamB | SharedLoadStreamB |
| The iterator for B to load from shared memory. More...
|
|
typedef GemmConfig::MultiplyAdd | MultiplyAdd |
| The multiply-add functor. More...
|
|
typedef GemmEpilogue< GemmEpilogueTraits_ > | Epilogue |
| The epilogue. More...
|
|
typedef Epilogue::ScalarC | ScalarC |
| The scalars in the epilogue. More...
|
|
typedef Epilogue::ScalarD | ScalarD |
|
typedef IdentityBlockSwizzle | BlockSwizzle |
| The block swizzle to reorganize the grid. More...
|
|
typedef Index_ | Index |
| The index. More...
|
|
typedef ClearAccumulators< GemmConfig_::Accumulators::Element > | ClearAccumulators |
| Clear the accumulators. More...
|
|
typedef GlobalLoadStreamPair< GlobalLoadStreamA, GlobalLoadStreamB, GemmConfig::kResidueInProlog > | GlobalLoadStream |
| Assemble the global load streams for A/B. More...
|
|
typedef GlobalLoadStream::ThreadblockTileStorage | ThreadblockTileStorage |
| Memory needed to store the threadblock-scoped GEMM tile. More...
|
|
typedef SharedStreamPair< SharedLoadStreamA, SharedLoadStreamB > | SharedStream |
| Assemble the shared load streams for A/B. More...
|
|
static CUTLASS_DEVICE void | shared_load_fence (bool in_loop) |
| The memory fence for shared loads. More...
|
|
static CUTLASS_DEVICE void | shared_store_fence (bool in_loop) |
| The memory fence for shared stores. More...
|
|
static MatrixLayout::Kind const | kLayoutA |
| The layout of A. More...
|
|
static MatrixLayout::Kind const | kLayoutB |
| The layout of B. More...
|
|