GemmHostArgs Struct Reference

GemmHostArgs Struct Reference#

Composable Kernel: ck_tile::GemmHostArgs Struct Reference
ck_tile::GemmHostArgs Struct Reference

#include <gemm_kernel.hpp>

Inheritance diagram for ck_tile::GemmHostArgs:
ck_tile::GemmProblem ck_tile::BatchedGemmHostArgs ck_tile::GroupedGemmHostArgs

Public Member Functions

CK_TILE_HOST GemmHostArgs ()=default
 
CK_TILE_HOST GemmHostArgs (const void *a_ptr_, const void *b_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_, index_t stride_C_)
 
- Public Member Functions inherited from ck_tile::GemmProblem
CK_TILE_HOST GemmProblem ()=default
 
CK_TILE_HOST GemmProblem (index_t M_, index_t N_, index_t K_, index_t stride_A_, index_t stride_B_, index_t stride_C_)
 

Public Attributes

const void * a_ptr
 
const void * b_ptr
 
void * c_ptr
 
index_t k_batch
 
- Public Attributes inherited from ck_tile::GemmProblem
index_t M
 
index_t N
 
index_t K
 
index_t stride_A
 
index_t stride_B
 
index_t stride_C
 

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 *  c_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_C_ 
)
inline

Member Data Documentation

◆ 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

◆ k_batch

index_t ck_tile::GemmHostArgs::k_batch

The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/ops/gemm/kernel/gemm_kernel.hpp