MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor > Struct Template Reference

MaskedGroupedFlatmmHostArgs&lt; ScaleM, ScaleN, NumDTensor &gt; Struct Template Reference#

Composable Kernel: ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor > Struct Template Reference
Public Member Functions | Public Attributes | List of all members
ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor > Struct Template Reference

#include <grouped_flatmm_kernel.hpp>

Public Member Functions

CK_TILE_HOST MaskedGroupedFlatmmHostArgs ()=default
CK_TILE_HOST MaskedGroupedFlatmmHostArgs (index_t *M_indices_, index_t group_count_, index_t Max_M_, index_t N_, index_t K_, const void *a_ptr_, index_t stride_A_, const void *b_shuffle_ptr_, index_t stride_B_, const std::array< const void *, NumDTensor > &ds_ptr_, const std::array< index_t, NumDTensor > &stride_Ds_, void *c_ptr_, index_t stride_C_, index_t k_batch_, ScaleM scale_m_=nullptr, ScaleN scale_n_=nullptr)

Public Attributes

index_t * M_indices
index_t group_count
index_t M
index_t N
index_t K
const void * a_ptr
index_t stride_A
const void * b_shuffle_ptr
index_t stride_B
const std::array< const void *, NumDTensor > ds_ptr
const std::array< index_t, NumDTensor > stride_Ds
union { 
   void *   e_ptr 
   void *   c_ptr 
}; 
index_t stride_C
index_t k_batch
ScaleM scale_m = nullptr
ScaleN scale_n = nullptr

Constructor & Destructor Documentation

◆ MaskedGroupedFlatmmHostArgs() [1/2]

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
CK_TILE_HOST ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::MaskedGroupedFlatmmHostArgs ( )
default

◆ MaskedGroupedFlatmmHostArgs() [2/2]

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
CK_TILE_HOST ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::MaskedGroupedFlatmmHostArgs ( index_t * M_indices_,
index_t group_count_,
index_t Max_M_,
index_t N_,
index_t K_,
const void * a_ptr_,
index_t stride_A_,
const void * b_shuffle_ptr_,
index_t stride_B_,
const std::array< const void *, NumDTensor > & ds_ptr_,
const std::array< index_t, NumDTensor > & stride_Ds_,
void * c_ptr_,
index_t stride_C_,
index_t k_batch_,
ScaleM scale_m_ = nullptr,
ScaleN scale_n_ = nullptr )
inline

Member Data Documentation

◆ [union]

union { ... } ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >

◆ a_ptr

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
const void* ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::a_ptr

◆ b_shuffle_ptr

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
const void* ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::b_shuffle_ptr

◆ c_ptr

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
void* ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::c_ptr

◆ ds_ptr

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
const std::array<const void*, NumDTensor> ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::ds_ptr

◆ e_ptr

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
void* ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::e_ptr

◆ group_count

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
index_t ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::group_count

◆ K

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
index_t ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::K

◆ k_batch

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
index_t ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::k_batch

◆ M

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
index_t ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::M

◆ M_indices

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
index_t* ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::M_indices

◆ N

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
index_t ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::N

◆ scale_m

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
ScaleM ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::scale_m = nullptr

◆ scale_n

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
ScaleN ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::scale_n = nullptr

◆ stride_A

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
index_t ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::stride_A

◆ stride_B

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
index_t ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::stride_B

◆ stride_C

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
index_t ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::stride_C

◆ stride_Ds

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
const std::array<index_t, NumDTensor> ck_tile::MaskedGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::stride_Ds
The documentation for this struct was generated from the following file: