TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec > Struct Template Reference

TransformBatchedContractionContractionToBatchedGemmGemm&lt; NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec &gt; Struct Template Reference#

Composable Kernel: ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec > Struct Template Reference
Static Public Member Functions | Static Public Attributes | List of all members
ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec > Struct Template Reference

#include <transform_contraction_to_gemm.hpp>

Static Public Member Functions

static auto MakeAGridDescriptorPair (const std::vector< index_t > &a_gs_ms_ks_lengths_vec, const std::vector< index_t > &a_gs_ms_ks_strides_vec)
static auto MakeAGridDescriptor_G_M_K (const std::vector< index_t > &a_gs_ms_ks_lengths_vec, const std::vector< index_t > &a_gs_ms_ks_strides_vec)
static auto MakeAGridDescriptor_M_K (const std::vector< index_t > &a_gs_ms_ks_lengths_vec, const std::vector< index_t > &a_gs_ms_ks_strides_vec)
template<typename AGridDesc_M_K, typename Number>
__host__ static __device__ constexpr auto MakeAGridDescriptor_AK0_M_AK1 (const AGridDesc_M_K &a_grid_desc_m_k, const Number &AK1)
static auto MakeB0GridDescriptorPair (const std::vector< index_t > &b0_gs_ns_ks_lengths_vec, const std::vector< index_t > &b0_gs_ns_ks_strides_vec)
static auto MakeB0GridDescriptor_G_N_K (const std::vector< index_t > &b0_gs_ns_ks_lengths_vec, const std::vector< index_t > &b0_gs_ns_ks_strides_vec)
static auto MakeB0GridDescriptor_N_K (const std::vector< index_t > &b0_gs_ns_ks_lengths_vec, const std::vector< index_t > &b0_gs_ns_ks_strides_vec)
template<typename BGridDesc_N_K, typename Number>
__host__ static __device__ constexpr auto MakeB0GridDescriptor_BK0_N_BK1 (const BGridDesc_N_K &b_grid_desc_n_k, const Number &BK1)
static auto MakeB1GridDescriptorPair (const std::vector< index_t > &b1_gs_os_ns_lengths_vec, const std::vector< index_t > &b1_gs_os_ns_strides_vec)
static auto MakeB1GridDescriptor_G_N_K (const std::vector< index_t > &b1_gs_os_ns_lengths_vec, const std::vector< index_t > &b1_gs_os_ns_strides_vec)
static auto MakeB1GridDescriptor_N_K (const std::vector< index_t > &b1_gs_os_ns_lengths_vec, const std::vector< index_t > &b1_gs_os_ns_strides_vec)
template<typename B1GridDesc_N_K, typename Number>
__host__ static __device__ constexpr auto MakeB1GridDescriptor_BK0_N_BK1 (const B1GridDesc_N_K &b1_grid_desc_n_k, const Number &B1K1)
static auto MakeCGridDescriptorPair (const std::vector< index_t > &c_gs_ms_os_lengths_vec, const std::vector< index_t > &c_gs_ms_os_strides_vec)
static auto MakeCGridDescriptor_G_M_N (const std::vector< index_t > &c_gs_ms_os_lengths_vec, const std::vector< index_t > &c_gs_ms_os_strides_vec)
static auto MakeCGridDescriptor_M_N (const std::vector< index_t > &c_gs_ms_os_lengths_vec, const std::vector< index_t > &c_gs_ms_os_strides_vec)

Static Public Attributes

static constexpr auto I0 = Number<0>{}
static constexpr auto I1 = Number<1>{}
static constexpr auto I2 = Number<2>{}
static constexpr auto I3 = Number<3>{}
static constexpr auto I4 = Number<4>{}
static constexpr index_t NumDimG = NumDims_G_M_N_K_O::At(I0)
static constexpr index_t NumDimM = NumDims_G_M_N_K_O::At(I1)
static constexpr index_t NumDimN = NumDims_G_M_N_K_O::At(I2)
static constexpr index_t NumDimK = NumDims_G_M_N_K_O::At(I3)
static constexpr index_t NumDimO = NumDims_G_M_N_K_O::At(I4)
static constexpr index_t MPerBlock = PerBlock_M_N_K_O::At(I0)
static constexpr index_t NPerBlock = PerBlock_M_N_K_O::At(I1)
static constexpr index_t KPerBlock = PerBlock_M_N_K_O::At(I2)
static constexpr index_t OPerBlock = PerBlock_M_N_K_O::At(I3)
static constexpr auto matrix_padder

Member Function Documentation

◆ MakeAGridDescriptor_AK0_M_AK1()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
template<typename AGridDesc_M_K, typename Number>
__host__ static __device__ constexpr auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeAGridDescriptor_AK0_M_AK1 ( const AGridDesc_M_K & a_grid_desc_m_k,
const Number & AK1 )
inlinestaticconstexpr

◆ MakeAGridDescriptor_G_M_K()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeAGridDescriptor_G_M_K ( const std::vector< index_t > & a_gs_ms_ks_lengths_vec,
const std::vector< index_t > & a_gs_ms_ks_strides_vec )
inlinestatic

◆ MakeAGridDescriptor_M_K()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeAGridDescriptor_M_K ( const std::vector< index_t > & a_gs_ms_ks_lengths_vec,
const std::vector< index_t > & a_gs_ms_ks_strides_vec )
inlinestatic

◆ MakeAGridDescriptorPair()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeAGridDescriptorPair ( const std::vector< index_t > & a_gs_ms_ks_lengths_vec,
const std::vector< index_t > & a_gs_ms_ks_strides_vec )
inlinestatic

◆ MakeB0GridDescriptor_BK0_N_BK1()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
template<typename BGridDesc_N_K, typename Number>
__host__ static __device__ constexpr auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeB0GridDescriptor_BK0_N_BK1 ( const BGridDesc_N_K & b_grid_desc_n_k,
const Number & BK1 )
inlinestaticconstexpr

◆ MakeB0GridDescriptor_G_N_K()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeB0GridDescriptor_G_N_K ( const std::vector< index_t > & b0_gs_ns_ks_lengths_vec,
const std::vector< index_t > & b0_gs_ns_ks_strides_vec )
inlinestatic

◆ MakeB0GridDescriptor_N_K()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeB0GridDescriptor_N_K ( const std::vector< index_t > & b0_gs_ns_ks_lengths_vec,
const std::vector< index_t > & b0_gs_ns_ks_strides_vec )
inlinestatic

◆ MakeB0GridDescriptorPair()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeB0GridDescriptorPair ( const std::vector< index_t > & b0_gs_ns_ks_lengths_vec,
const std::vector< index_t > & b0_gs_ns_ks_strides_vec )
inlinestatic

◆ MakeB1GridDescriptor_BK0_N_BK1()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
template<typename B1GridDesc_N_K, typename Number>
__host__ static __device__ constexpr auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeB1GridDescriptor_BK0_N_BK1 ( const B1GridDesc_N_K & b1_grid_desc_n_k,
const Number & B1K1 )
inlinestaticconstexpr

◆ MakeB1GridDescriptor_G_N_K()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeB1GridDescriptor_G_N_K ( const std::vector< index_t > & b1_gs_os_ns_lengths_vec,
const std::vector< index_t > & b1_gs_os_ns_strides_vec )
inlinestatic

◆ MakeB1GridDescriptor_N_K()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeB1GridDescriptor_N_K ( const std::vector< index_t > & b1_gs_os_ns_lengths_vec,
const std::vector< index_t > & b1_gs_os_ns_strides_vec )
inlinestatic

◆ MakeB1GridDescriptorPair()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeB1GridDescriptorPair ( const std::vector< index_t > & b1_gs_os_ns_lengths_vec,
const std::vector< index_t > & b1_gs_os_ns_strides_vec )
inlinestatic

◆ MakeCGridDescriptor_G_M_N()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeCGridDescriptor_G_M_N ( const std::vector< index_t > & c_gs_ms_os_lengths_vec,
const std::vector< index_t > & c_gs_ms_os_strides_vec )
inlinestatic

◆ MakeCGridDescriptor_M_N()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeCGridDescriptor_M_N ( const std::vector< index_t > & c_gs_ms_os_lengths_vec,
const std::vector< index_t > & c_gs_ms_os_strides_vec )
inlinestatic

◆ MakeCGridDescriptorPair()

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MakeCGridDescriptorPair ( const std::vector< index_t > & c_gs_ms_os_lengths_vec,
const std::vector< index_t > & c_gs_ms_os_strides_vec )
inlinestatic

Member Data Documentation

◆ I0

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::I0 = Number<0>{}
staticconstexpr

◆ I1

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::I1 = Number<1>{}
staticconstexpr

◆ I2

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::I2 = Number<2>{}
staticconstexpr

◆ I3

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::I3 = Number<3>{}
staticconstexpr

◆ I4

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::I4 = Number<4>{}
staticconstexpr

◆ KPerBlock

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
index_t ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::KPerBlock = PerBlock_M_N_K_O::At(I2)
staticconstexpr

◆ matrix_padder

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
auto ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::matrix_padder
staticconstexpr
Initial value:
=
device::GemmGemmPadder<GemmSpec, index_t, index_t, index_t, index_t>{
MPerBlock, NPerBlock, KPerBlock, OPerBlock}
ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm::NPerBlock
static constexpr index_t NPerBlock
Definition transform_contraction_to_gemm.hpp:135
ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm::MPerBlock
static constexpr index_t MPerBlock
Definition transform_contraction_to_gemm.hpp:134
ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm::OPerBlock
static constexpr index_t OPerBlock
Definition transform_contraction_to_gemm.hpp:137
ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm::KPerBlock
static constexpr index_t KPerBlock
Definition transform_contraction_to_gemm.hpp:136
ck::tensor_operation::device::GemmGemmPadder
Definition matrix_padder.hpp:63

◆ MPerBlock

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
index_t ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::MPerBlock = PerBlock_M_N_K_O::At(I0)
staticconstexpr

◆ NPerBlock

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
index_t ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::NPerBlock = PerBlock_M_N_K_O::At(I1)
staticconstexpr

◆ NumDimG

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
index_t ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::NumDimG = NumDims_G_M_N_K_O::At(I0)
staticconstexpr

◆ NumDimK

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
index_t ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::NumDimK = NumDims_G_M_N_K_O::At(I3)
staticconstexpr

◆ NumDimM

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
index_t ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::NumDimM = NumDims_G_M_N_K_O::At(I1)
staticconstexpr

◆ NumDimN

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
index_t ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::NumDimN = NumDims_G_M_N_K_O::At(I2)
staticconstexpr

◆ NumDimO

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
index_t ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::NumDimO = NumDims_G_M_N_K_O::At(I4)
staticconstexpr

◆ OPerBlock

template<typename NumDims_G_M_N_K_O, typename PerBlock_M_N_K_O, device::GemmSpecialization GemmSpec, device::TensorSpecialization ASpec, device::TensorSpecialization B0Spec, device::TensorSpecialization B1Spec, device::TensorSpecialization CSpec>
index_t ck::tensor_operation::TransformBatchedContractionContractionToBatchedGemmGemm< NumDims_G_M_N_K_O, PerBlock_M_N_K_O, GemmSpec, ASpec, B0Spec, B1Spec, CSpec >::OPerBlock = PerBlock_M_N_K_O::At(I3)
staticconstexpr
The documentation for this struct was generated from the following file: