GemmHostArgs Struct Reference

GemmHostArgs Struct Reference#

Composable Kernel: ck_tile::GemmHostArgs Struct Reference

The GEMM kernel host arguments. More...

#include <gemm_kernel.hpp>

Public Member Functions
CK_TILE_HOST	GemmHostArgs ()=default

CK_TILE_HOST	GemmHostArgs (const void a_ptr_, const void b_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_, index_t stride_E_)

Public Attributes
const void *	a_ptr

const void *	b_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

union {
index_t stride_E

index_t stride_C

};

index_t	k_batch

Detailed Description

The GEMM kernel host arguments.

Overview: This structure is passed to GemmKernel when creating kernel arguments object. It contain all necessary information required to build proper kernel argument and launch kernel on GPU. This structure defines the GEMM problem configuration by stating all required information like M,N,K sizes and respective strides.

Constructor & Destructor Documentation

◆ GemmHostArgs() [1/2]

CK_TILE_HOST ck_tile::GemmHostArgs::GemmHostArgs ( )

default

◆ GemmHostArgs() [2/2]

CK_TILE_HOST ck_tile::GemmHostArgs::GemmHostArgs	(	const void *	a_ptr_,
		const void *	b_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_,
		index_t	stride_E_
	)

inline

Member Data Documentation

◆

union { ... }

◆

union { ... }

◆ a_ptr

const void* ck_tile::GemmHostArgs::a_ptr

◆ b_ptr

const void* ck_tile::GemmHostArgs::b_ptr

◆ c_ptr

void* ck_tile::GemmHostArgs::c_ptr

◆ e_ptr

void* ck_tile::GemmHostArgs::e_ptr

◆ K

index_t ck_tile::GemmHostArgs::K

◆ k_batch

index_t ck_tile::GemmHostArgs::k_batch

◆ M

index_t ck_tile::GemmHostArgs::M

◆ N

index_t ck_tile::GemmHostArgs::N

◆ stride_A

index_t ck_tile::GemmHostArgs::stride_A

◆ stride_B

index_t ck_tile::GemmHostArgs::stride_B

◆ stride_C

index_t ck_tile::GemmHostArgs::stride_C

◆ stride_E

index_t ck_tile::GemmHostArgs::stride_E

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

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/gemm/kernel/gemm_kernel.hpp