FlatmmHostArgs Struct Reference

FlatmmHostArgs Struct Reference#

Composable Kernel: ck_tile::FlatmmHostArgs Struct Reference
ck_tile::FlatmmHostArgs Struct Reference

#include <flatmm_kernel.hpp>

Inheritance diagram for ck_tile::FlatmmHostArgs:
ck_tile::FlatmmProblem

Public Member Functions

CK_TILE_HOST FlatmmHostArgs ()=default
 
CK_TILE_HOST FlatmmHostArgs (const void *a_ptr_, const void *b_shuffle_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::FlatmmProblem
CK_TILE_HOST FlatmmProblem ()=default
 
CK_TILE_HOST FlatmmProblem (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_shuffle_ptr
 
void * c_ptr
 
index_t k_batch
 
- Public Attributes inherited from ck_tile::FlatmmProblem
index_t M
 
index_t N
 
index_t K
 
index_t stride_A
 
index_t stride_B
 
index_t stride_C
 

Constructor & Destructor Documentation

◆ FlatmmHostArgs() [1/2]

CK_TILE_HOST ck_tile::FlatmmHostArgs::FlatmmHostArgs ( )
default

◆ FlatmmHostArgs() [2/2]

CK_TILE_HOST ck_tile::FlatmmHostArgs::FlatmmHostArgs ( const void *  a_ptr_,
const void *  b_shuffle_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::FlatmmHostArgs::a_ptr

◆ b_shuffle_ptr

const void* ck_tile::FlatmmHostArgs::b_shuffle_ptr

◆ c_ptr

void* ck_tile::FlatmmHostArgs::c_ptr

◆ k_batch

index_t ck_tile::FlatmmHostArgs::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-7.0.0/include/ck_tile/ops/flatmm/kernel/flatmm_kernel.hpp