docs/structcutlass_1_1gemm_1_1threadblock_1_1MmaPolicy.html
| | CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers |
Public Types | Static Public Attributes | List of all members
cutlass::gemm::threadblock::MmaPolicy< Operator_, SmemPaddingA_, SmemPaddingB_, PartitionsK > Struct Template Reference
Policy object describing MmaTensorOp.
#include <mma_base.h>
|
|
| using | Operator = Operator_ |
| | Warp-level GEMM operator (concept: gemm::warp::MmaTensorOp or gemm::warp::MmaSimt) More...
|
| |
| using | SmemPaddingA = SmemPaddingA_ |
| | Padding used for A operand in shared memory. More...
|
| |
| using | SmemPaddingB = SmemPaddingB_ |
| | Padding used for B operand in shared memory. More...
|
| |
|
|
| static int const | kPartitionsK = PartitionsK |
| | Number of partitions of K dimension. More...
|
| |
template<typename Operator_ , typename SmemPaddingA_ , typename SmemPaddingB_ , int PartitionsK = 1>
| using cutlass::gemm::threadblock::MmaPolicy< Operator_, SmemPaddingA_, SmemPaddingB_, PartitionsK >::Operator = Operator_ |
template<typename Operator_ , typename SmemPaddingA_ , typename SmemPaddingB_ , int PartitionsK = 1>
| using cutlass::gemm::threadblock::MmaPolicy< Operator_, SmemPaddingA_, SmemPaddingB_, PartitionsK >::SmemPaddingA = SmemPaddingA_ |
template<typename Operator_ , typename SmemPaddingA_ , typename SmemPaddingB_ , int PartitionsK = 1>
| using cutlass::gemm::threadblock::MmaPolicy< Operator_, SmemPaddingA_, SmemPaddingB_, PartitionsK >::SmemPaddingB = SmemPaddingB_ |
template<typename Operator_ , typename SmemPaddingA_ , typename SmemPaddingB_ , int PartitionsK = 1>
|
| int const cutlass::gemm::threadblock::MmaPolicy< Operator_, SmemPaddingA_, SmemPaddingB_, PartitionsK >::kPartitionsK = PartitionsK |
| static |
The documentation for this struct was generated from the following file:
Generated by 1.8.11