FlatmmKernelArgs< NumDTensor > Struct Template Reference

• ck_tile
• FlatmmKernelArgs

Public Attributes | List of all members

ck_tile::FlatmmKernelArgs< 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

Member Data Documentation

◆ a_ptr

template<index_t NumDTensor = 0>

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

◆ b_ptr

template<index_t NumDTensor = 0>

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

◆ ds_ptr

template<index_t NumDTensor = 0>

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

◆ e_ptr

template<index_t NumDTensor = 0>

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

◆ K

template<index_t NumDTensor = 0>

index_t ck_tile::FlatmmKernelArgs< NumDTensor >::K

◆ k_batch

template<index_t NumDTensor = 0>

index_t ck_tile::FlatmmKernelArgs< NumDTensor >::k_batch

◆ M

template<index_t NumDTensor = 0>

index_t ck_tile::FlatmmKernelArgs< NumDTensor >::M

◆ N

template<index_t NumDTensor = 0>

index_t ck_tile::FlatmmKernelArgs< NumDTensor >::N

◆ stride_A

template<index_t NumDTensor = 0>

index_t ck_tile::FlatmmKernelArgs< NumDTensor >::stride_A

◆ stride_B

template<index_t NumDTensor = 0>

index_t ck_tile::FlatmmKernelArgs< NumDTensor >::stride_B

◆ stride_Ds

template<index_t NumDTensor = 0>

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

◆ stride_E

template<index_t NumDTensor = 0>

index_t ck_tile::FlatmmKernelArgs< NumDTensor >::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/flatmm/kernel/flatmm_kernel.hpp