FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor > Struct Template Reference

• ck_tile
• FlatmmKernelArgs

Public Attributes | List of all members

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

#include <flatmm_kernel.hpp>

Public Attributes
const void *	a_ptr
const void *	b_ptr
const std::array< const void *, NumDTensor >	ds_ptr
void *	e_ptr
index_t	M
index_t	N
index_t	K
index_t	stride_A
index_t	stride_B
std::array< index_t, NumDTensor >	stride_Ds
index_t	stride_E
index_t	k_batch
ScaleM	scale_m_ptr = nullptr
ScaleN	scale_n_ptr = nullptr

Member Data Documentation

◆ a_ptr

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

const void* ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::a_ptr

◆ b_ptr

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

const void* ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::b_ptr

◆ ds_ptr

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

const std::array<const void*, NumDTensor> ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::ds_ptr

◆ e_ptr

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

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

◆ K

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

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

◆ k_batch

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

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

◆ M

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

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

◆ N

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

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

◆ scale_m_ptr

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

ScaleM ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::scale_m_ptr = nullptr

◆ scale_n_ptr

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

ScaleN ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::scale_n_ptr = nullptr

◆ stride_A

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

index_t ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::stride_A

◆ stride_B

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

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

◆ stride_Ds

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

std::array<index_t, NumDTensor> ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::stride_Ds

◆ stride_E

template<class ScaleM, class ScaleN, index_t NumDTensor = 0>

index_t ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::stride_E

---

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

• flatmm_kernel.hpp