GEMM problem description.
#include <gemm_desc.h>
|
CUTLASS_HOST_DEVICE | GemmDesc () |
| Default ctor. More...
|
|
CUTLASS_HOST_DEVICE | GemmDesc (Coord< 3 > _problem_size, SType _alpha, TensorRefA const &_A, TensorRefB const &_B, SType _beta, TensorRefC const &_C, TensorRefD const &_D) |
| Constructor for basic GEMM with batch count = 1. More...
|
|
CUTLASS_HOST_DEVICE | GemmDesc (GemmCoord _problem_size, SType _alpha, TensorRefA const &_A, TensorRefB const &_B, SType _beta, TensorRefC const &_C, TensorRefD const &_D) |
| Constructor for basic GEMM with batch count = 1. More...
|
|
CUTLASS_HOST_DEVICE | GemmDesc (GemmCoord _problem_size, SType _alpha, TensorRefA const &_A, long long _batch_stride_A, TensorRefB const &_B, long long _batch_stride_B, SType _beta, TensorRefC const &_C, long long _batch_stride_C, TensorRefD const &_D, long long _batch_stride_D) |
| Constructor for strided batch GEMM GEMM. More...
|
|
◆ AType
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ BType
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ CType
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ DType
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ Index
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ SType
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ TensorRefA
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ TensorRefB
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ TensorRefC
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ TensorRefD
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ GemmDesc() [1/4]
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ GemmDesc() [2/4]
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
CUTLASS_HOST_DEVICE cutlass::gemm::GemmDesc< AType_, BType_, CType_, DType_, SType_, Index_ >::GemmDesc |
( |
Coord< 3 > |
_problem_size, |
|
|
SType |
_alpha, |
|
|
TensorRefA const & |
_A, |
|
|
TensorRefB const & |
_B, |
|
|
SType |
_beta, |
|
|
TensorRefC const & |
_C, |
|
|
TensorRefD const & |
_D |
|
) |
| |
|
inline |
◆ GemmDesc() [3/4]
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
CUTLASS_HOST_DEVICE cutlass::gemm::GemmDesc< AType_, BType_, CType_, DType_, SType_, Index_ >::GemmDesc |
( |
GemmCoord |
_problem_size, |
|
|
SType |
_alpha, |
|
|
TensorRefA const & |
_A, |
|
|
TensorRefB const & |
_B, |
|
|
SType |
_beta, |
|
|
TensorRefC const & |
_C, |
|
|
TensorRefD const & |
_D |
|
) |
| |
|
inline |
◆ GemmDesc() [4/4]
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
CUTLASS_HOST_DEVICE cutlass::gemm::GemmDesc< AType_, BType_, CType_, DType_, SType_, Index_ >::GemmDesc |
( |
GemmCoord |
_problem_size, |
|
|
SType |
_alpha, |
|
|
TensorRefA const & |
_A, |
|
|
long long |
_batch_stride_A, |
|
|
TensorRefB const & |
_B, |
|
|
long long |
_batch_stride_B, |
|
|
SType |
_beta, |
|
|
TensorRefC const & |
_C, |
|
|
long long |
_batch_stride_C, |
|
|
TensorRefD const & |
_D, |
|
|
long long |
_batch_stride_D |
|
) |
| |
|
inline |
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ alpha
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ batch_stride_A
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ batch_stride_B
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ batch_stride_C
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ batch_stride_D
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ beta
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
◆ problem_size
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int>
The documentation for this struct was generated from the following file: