#include <device_gemm_traits.h>
|
| Params () |
| ctor More...
|
|
| Params (Index m_, Index n_, Index k_) |
| ctor More...
|
|
void | init_problem (Index m_, Index n_, Index k_) |
| init problem is needed if using default ctor More...
|
|
int | initialize (Scalar alpha_, ScalarA const *d_a_, Index lda_, ScalarB const *d_b_, Index ldb_, Scalar beta_, ScalarC const *d_c_, Index ldc_, ScalarD *d_d_, Index ldd_, ScalarAccum *workspace_ptr_) |
|
int | required_workspace_memory_in_byte () |
|
◆ Params() [1/2]
template<typename GemmTraits_ , typename ReductionTraits_ >
◆ Params() [2/2]
template<typename GemmTraits_ , typename ReductionTraits_ >
◆ init_problem()
template<typename GemmTraits_ , typename ReductionTraits_ >
◆ initialize()
template<typename GemmTraits_ , typename ReductionTraits_ >
int cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::Params::initialize |
( |
Scalar |
alpha_, |
|
|
ScalarA const * |
d_a_, |
|
|
Index |
lda_, |
|
|
ScalarB const * |
d_b_, |
|
|
Index |
ldb_, |
|
|
Scalar |
beta_, |
|
|
ScalarC const * |
d_c_, |
|
|
Index |
ldc_, |
|
|
ScalarD * |
d_d_, |
|
|
Index |
ldd_, |
|
|
ScalarAccum * |
workspace_ptr_ |
|
) |
| |
|
inline |
◆ required_workspace_memory_in_byte()
template<typename GemmTraits_ , typename ReductionTraits_ >
◆ GemmParams
template<typename GemmTraits_ , typename ReductionTraits_ >
◆ problem_size
template<typename GemmTraits_ , typename ReductionTraits_ >
◆ problem_size_initialized
template<typename GemmTraits_ , typename ReductionTraits_ >
◆ ReductionParams
template<typename GemmTraits_ , typename ReductionTraits_ >
◆ workspace_ptr
template<typename GemmTraits_ , typename ReductionTraits_ >
◆ workspace_size
template<typename GemmTraits_ , typename ReductionTraits_ >
The documentation for this struct was generated from the following file: