ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor > Struct Template Reference

• ck_tile
• ContiguousGroupedFlatmmHostArgs

Public Member Functions | Public Attributes | List of all members

ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor > Struct Template Reference

#include <grouped_flatmm_kernel.hpp>

Public Member Functions
CK_TILE_HOST	ContiguousGroupedFlatmmHostArgs ()=default
CK_TILE_HOST	ContiguousGroupedFlatmmHostArgs (index_t *M_indices_, 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_indices
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

◆ ContiguousGroupedFlatmmHostArgs() [1/2]

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

CK_TILE_HOST ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::ContiguousGroupedFlatmmHostArgs ( )

default

◆ ContiguousGroupedFlatmmHostArgs() [2/2]

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

CK_TILE_HOST ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::ContiguousGroupedFlatmmHostArgs ( index_t * M_indices_, 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::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >

◆ a_ptr

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

const void* ck_tile::ContiguousGroupedFlatmmHostArgs< 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::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::b_shuffle_ptr

◆ c_ptr

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

void* ck_tile::ContiguousGroupedFlatmmHostArgs< 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::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::ds_ptr

◆ e_ptr

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

void* ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::e_ptr

◆ group_count

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

index_t ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::group_count

◆ K

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

index_t ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::K

◆ k_batch

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

index_t ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::k_batch

◆ M

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

index_t ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::M

◆ M_indices

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

index_t* ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::M_indices

◆ N

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

index_t ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::N

◆ scale_m

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

ScaleM ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::scale_m = nullptr

◆ scale_n

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

ScaleN ck_tile::ContiguousGroupedFlatmmHostArgs< 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::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::stride_A

◆ stride_B

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

index_t ck_tile::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::stride_B

◆ stride_C

template<class ScaleM = FlatmmScalePointer<-1>, class ScaleN = FlatmmScalePointer<-1>, index_t NumDTensor = 0>

index_t ck_tile::ContiguousGroupedFlatmmHostArgs< 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::ContiguousGroupedFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::stride_Ds

---

The documentation for this struct was generated from the following file:

• grouped_flatmm_kernel.hpp