ScaleFlatmmHostArgs< ScaleM, ScaleN, NumDTensor > Struct Template Reference

• ck_tile
• ScaleFlatmmHostArgs

Public Member Functions | Public Attributes | List of all members

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

#include <flatmm_kernel.hpp>

Inheritance diagram for ck_tile::ScaleFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >:

Public Member Functions
CK_TILE_HOST	ScaleFlatmmHostArgs ()=default
CK_TILE_HOST	ScaleFlatmmHostArgs (const void *a_ptr_, const void *b_shuffle_ptr_, const std::array< const void *, NumDTensor > &ds_ptr_, void *c_ptr_, index_t k_batch_, index_t M_, index_t N_, index_t K_, index_t stride_A_, index_t stride_B_, const std::array< index_t, NumDTensor > &stride_Ds_, index_t stride_C_, ScaleM scale_m_=nullptr, ScaleN scale_n_=nullptr)
Public Member Functions inherited from ck_tile::BaseFlatmmHostArgs< NumDTensor >
CK_TILE_HOST	BaseFlatmmHostArgs ()=default
CK_TILE_HOST	BaseFlatmmHostArgs (const void *a_ptr_, const void *b_ptr_, const std::array< const void *, NumDTensor > &ds_ptr_, void *e_ptr_, index_t k_batch_, index_t M_, index_t N_, index_t K_, index_t stride_A_, index_t stride_B_, const std::array< index_t, NumDTensor > &stride_Ds_, index_t stride_E_)

Public Attributes
ScaleM	scale_m = nullptr
ScaleN	scale_n = nullptr
Public Attributes inherited from ck_tile::BaseFlatmmHostArgs< NumDTensor >
const void *	a_ptr
const void *	b_ptr
const std::array< const void *, NumDTensor >	ds_ptr
union {
void *   e_ptr
void *   c_ptr
};
index_t	M
index_t	N
index_t	K
index_t	stride_A
index_t	stride_B
const std::array< index_t, NumDTensor >	stride_Ds
union {
index_t   stride_E
index_t   stride_C
};
index_t	k_batch

Constructor & Destructor Documentation

◆ ScaleFlatmmHostArgs() [1/2]

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

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

default

◆ ScaleFlatmmHostArgs() [2/2]

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

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

inline

Member Data Documentation

◆ scale_m

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

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

◆ scale_n

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

ScaleN ck_tile::ScaleFlatmmHostArgs< ScaleM, ScaleN, NumDTensor >::scale_n = nullptr

---

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

• flatmm_kernel.hpp