#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: