#include <flatmm_kernel.hpp>
|
CK_TILE_HOST | FlatmmHostArgs ()=default |
|
CK_TILE_HOST | FlatmmHostArgs (const void *a_ptr_, const void *b_ptr_, const std::array< const void *, NumDTensor > &ds_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_, const std::array< index_t, NumDTensor > &stride_Ds_, index_t stride_E_) |
|
◆ FlatmmHostArgs() [1/2]
template<index_t NumDTensor = 0>
◆ FlatmmHostArgs() [2/2]
template<index_t NumDTensor = 0>
CK_TILE_HOST ck_tile::FlatmmHostArgs< NumDTensor >::FlatmmHostArgs |
( |
const void * |
a_ptr_, |
|
|
const void * |
b_ptr_, |
|
|
const std::array< const void *, NumDTensor > & |
ds_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_, |
|
|
const std::array< index_t, NumDTensor > & |
stride_Ds_, |
|
|
index_t |
stride_E_ |
|
) |
| |
|
inline |
◆ a_ptr
template<index_t NumDTensor = 0>
◆ b_ptr
template<index_t NumDTensor = 0>
◆ c_ptr
template<index_t NumDTensor = 0>
◆ ds_ptr
template<index_t NumDTensor = 0>
◆ e_ptr
template<index_t NumDTensor = 0>
template<index_t NumDTensor = 0>
◆ k_batch
template<index_t NumDTensor = 0>
template<index_t NumDTensor = 0>
template<index_t NumDTensor = 0>
◆ stride_A
template<index_t NumDTensor = 0>
◆ stride_B
template<index_t NumDTensor = 0>
◆ stride_C
template<index_t NumDTensor = 0>
◆ stride_Ds
template<index_t NumDTensor = 0>
◆ stride_E
template<index_t NumDTensor = 0>
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