BaseFlatmmHostArgs< NumDTensor > Struct Template Reference

BaseFlatmmHostArgs< NumDTensor > Struct Template Reference#

Composable Kernel: ck_tile::BaseFlatmmHostArgs< NumDTensor > Struct Template Reference

#include <flatmm_kernel.hpp>

Inheritance diagram for ck_tile::BaseFlatmmHostArgs< NumDTensor >:

Public Member Functions
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
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

◆ BaseFlatmmHostArgs() [1/2]

template<index_t NumDTensor = 0>

CK_TILE_HOST ck_tile::BaseFlatmmHostArgs< NumDTensor >::BaseFlatmmHostArgs ( )

default

◆ BaseFlatmmHostArgs() [2/2]

template<index_t NumDTensor = 0>

CK_TILE_HOST ck_tile::BaseFlatmmHostArgs< NumDTensor >::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_ )

inline

Member Data Documentation

◆ [union]

union { ... } ck_tile::BaseFlatmmHostArgs< NumDTensor >

◆ [union]

union { ... } ck_tile::BaseFlatmmHostArgs< NumDTensor >

◆ a_ptr

template<index_t NumDTensor = 0>

const void* ck_tile::BaseFlatmmHostArgs< NumDTensor >::a_ptr

◆ b_ptr

template<index_t NumDTensor = 0>

const void* ck_tile::BaseFlatmmHostArgs< NumDTensor >::b_ptr

◆ c_ptr

template<index_t NumDTensor = 0>

void* ck_tile::BaseFlatmmHostArgs< NumDTensor >::c_ptr

◆ ds_ptr

template<index_t NumDTensor = 0>

const std::array<const void*, NumDTensor> ck_tile::BaseFlatmmHostArgs< NumDTensor >::ds_ptr

◆ e_ptr

template<index_t NumDTensor = 0>

void* ck_tile::BaseFlatmmHostArgs< NumDTensor >::e_ptr

◆ K

template<index_t NumDTensor = 0>

index_t ck_tile::BaseFlatmmHostArgs< NumDTensor >::K

◆ k_batch

template<index_t NumDTensor = 0>

index_t ck_tile::BaseFlatmmHostArgs< NumDTensor >::k_batch

◆ M

template<index_t NumDTensor = 0>

index_t ck_tile::BaseFlatmmHostArgs< NumDTensor >::M

◆ N

template<index_t NumDTensor = 0>

index_t ck_tile::BaseFlatmmHostArgs< NumDTensor >::N

◆ stride_A

template<index_t NumDTensor = 0>

index_t ck_tile::BaseFlatmmHostArgs< NumDTensor >::stride_A

◆ stride_B

template<index_t NumDTensor = 0>

index_t ck_tile::BaseFlatmmHostArgs< NumDTensor >::stride_B

◆ stride_C

template<index_t NumDTensor = 0>

index_t ck_tile::BaseFlatmmHostArgs< NumDTensor >::stride_C

◆ stride_Ds

template<index_t NumDTensor = 0>

const std::array<index_t, NumDTensor> ck_tile::BaseFlatmmHostArgs< NumDTensor >::stride_Ds

◆ stride_E

template<index_t NumDTensor = 0>

index_t ck_tile::BaseFlatmmHostArgs< NumDTensor >::stride_E

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

flatmm_kernel.hpp