FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor > Struct Template Reference

FlatmmKernelArgs&lt; ScaleM, ScaleN, NumDTensor &gt; Struct Template Reference#

Composable Kernel: ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor > Struct Template Reference
Public Attributes | List of all members
ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, 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
 
ScaleM scale_m_ptr = nullptr
 
ScaleN scale_n_ptr = nullptr
 

Member Data Documentation

◆ a_ptr

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
const void* ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::a_ptr

◆ b_ptr

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
const void* ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::b_ptr

◆ ds_ptr

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
const std::array<const void*, NumDTensor> ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::ds_ptr

◆ e_ptr

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
void* ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::e_ptr

◆ K

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
index_t ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::K

◆ k_batch

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
index_t ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::k_batch

◆ M

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
index_t ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::M

◆ N

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
index_t ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::N

◆ scale_m_ptr

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
ScaleM ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::scale_m_ptr = nullptr

◆ scale_n_ptr

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
ScaleN ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::scale_n_ptr = nullptr

◆ stride_A

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
index_t ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::stride_A

◆ stride_B

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
index_t ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::stride_B

◆ stride_Ds

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
std::array<index_t, NumDTensor> ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, NumDTensor >::stride_Ds

◆ stride_E

template<class ScaleM , class ScaleN , index_t NumDTensor = 0>
index_t ck_tile::FlatmmKernelArgs< ScaleM, ScaleN, 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