GroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor > Struct Template Reference

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

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

#include <grouped_flatmm_kernel.hpp>

Public Member Functions

CK_TILE_HOST GroupedFlatmmHostArgs ()=default
CK_TILE_HOST GroupedFlatmmHostArgs (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_, void **c_ptr_, index_t *stride_C_, index_t k_batch_, ScaleM *scale_m_=nullptr, ScaleN *scale_n_=nullptr)

Public Attributes

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

◆ GroupedFlatmmHostArgs() [1/2]

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

◆ GroupedFlatmmHostArgs() [2/2]

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
CK_TILE_HOST ck_tile::GroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::GroupedFlatmmHostArgs ( 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_,
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::GroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >

◆ a_ptr

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
const void** ck_tile::GroupedFlatmmHostArgs< 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::GroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::b_shuffle_ptr

◆ c_ptr

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
void** ck_tile::GroupedFlatmmHostArgs< 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::GroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::ds_ptr

◆ e_ptr

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

◆ group_count

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

◆ K

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

◆ k_batch

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

◆ M

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

◆ N

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

◆ scale_m

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

◆ scale_n

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
ScaleN* ck_tile::GroupedFlatmmHostArgs< 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::GroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::stride_A

◆ stride_B

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

◆ stride_C

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>
index_t* ck_tile::GroupedFlatmmHostArgs< 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::GroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::stride_Ds
The documentation for this struct was generated from the following file: