Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <device_gemm_traits.h>
Classes | |
struct | Params |
Public Types | |
typedef GemmTraits_ | GemmTraits |
typedef ReductionTraits_ | ReductionTraits |
typedef SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ > | This_ |
typedef cutlass::gemm::DeviceGemm< This_ > | KernelClass |
typedef GemmTraits::Index | Index |
typedef ReductionTraits::ScalarAlphaBeta | Scalar |
typedef GemmTraits::ScalarA | ScalarA |
typedef GemmTraits::ScalarB | ScalarB |
typedef GemmTraits::ScalarD | ScalarAccum |
typedef ReductionTraits::ScalarC | ScalarC |
typedef ReductionTraits::ScalarD | ScalarD |
Static Public Attributes | |
static MatrixLayout::Kind const | kLayoutA = GemmTraits::kLayoutA |
The layout of A. can be deduced from the layout set in batched gemm. More... | |
static MatrixLayout::Kind const | kLayoutB = GemmTraits::kLayoutB |
The layout of B. can be deduced from the layout set in batched gemm. More... | |
typedef GemmTraits_ cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::GemmTraits |
typedef GemmTraits::Index cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::Index |
typedef cutlass::gemm::DeviceGemm<This_> cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::KernelClass |
typedef ReductionTraits_ cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::ReductionTraits |
typedef ReductionTraits::ScalarAlphaBeta cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::Scalar |
typedef GemmTraits::ScalarA cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::ScalarA |
typedef GemmTraits::ScalarD cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::ScalarAccum |
typedef GemmTraits::ScalarB cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::ScalarB |
typedef ReductionTraits::ScalarC cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::ScalarC |
typedef ReductionTraits::ScalarD cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::ScalarD |
typedef SplitkPIGemmTraits<GemmTraits_, ReductionTraits_> cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::This_ |
|
static |
|
static |