![]() |
CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Description of all GEMM computations.
#include <library.h>


Public Member Functions | |
| GemmDescription (GemmKind gemm_kind=GemmKind::kGemm, TensorDescription const &A=TensorDescription(), TensorDescription const &B=TensorDescription(), TensorDescription const &C=TensorDescription(), NumericTypeID element_epilogue=NumericTypeID::kInvalid, SplitKMode split_k_mode=SplitKMode::kNone, ComplexTransform transform_A=ComplexTransform::kNone, ComplexTransform transform_B=ComplexTransform::kNone) | |
Public Member Functions inherited from cutlass::library::OperationDescription | |
| OperationDescription (char const *name="unknown", OperationKind kind=OperationKind::kInvalid, TileDescription const &tile_description=TileDescription()) | |
Public Attributes | |
| GemmKind | gemm_kind |
| Indicates the kind of GEMM performed. More... | |
| TensorDescription | A |
| Describes the A operand. More... | |
| TensorDescription | B |
| Describes the B operand. More... | |
| TensorDescription | C |
| Describes the source and destination matrices. More... | |
| NumericTypeID | element_epilogue |
| Describes the data type of the scalars passed to the epilogue. More... | |
| SplitKMode | split_k_mode |
| Describes the structure of parallel reductions. More... | |
| ComplexTransform | transform_A |
| Transformation on A operand. More... | |
| ComplexTransform | transform_B |
| Transformation on B operand. More... | |
Public Attributes inherited from cutlass::library::OperationDescription | |
| char const * | name |
| Unique identifier describing the operation. More... | |
| OperationKind | kind |
| Kind of operation. More... | |
| TileDescription | tile_description |
| Describes the tiled structure of a GEMM-like computation. More... | |
|
inline |
| TensorDescription cutlass::library::GemmDescription::A |
| TensorDescription cutlass::library::GemmDescription::B |
| TensorDescription cutlass::library::GemmDescription::C |
| NumericTypeID cutlass::library::GemmDescription::element_epilogue |
| GemmKind cutlass::library::GemmDescription::gemm_kind |
| SplitKMode cutlass::library::GemmDescription::split_k_mode |
| ComplexTransform cutlass::library::GemmDescription::transform_A |
| ComplexTransform cutlass::library::GemmDescription::transform_B |
1.8.11