GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor > Struct Template Reference

GroupedGemmMultiABDKernelArgument&lt; NumATensor, NumBTensor, NumDTensor &gt; Struct Template Reference#

Composable Kernel: ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor > Struct Template Reference
Public Attributes | List of all members
ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor > Struct Template Reference

#include <device_grouped_gemm_multi_abd_fixed_nk.hpp>

Public Attributes

std::array< const void *, NumATensor > p_as_grid
std::array< const void *, NumBTensor > p_bs_grid
std::array< const void *, NumDTensor > p_ds_grid
void * p_e_grid
index_t M
index_t N
index_t K
std::array< index_t, NumATensor > StrideAs
std::array< index_t, NumBTensor > StrideBs
std::array< index_t, NumDTensor > StrideDs
index_t StrideE

Member Data Documentation

◆ K

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
index_t ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::K

◆ M

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
index_t ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::M

◆ N

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
index_t ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::N

◆ p_as_grid

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
std::array<const void*, NumATensor> ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::p_as_grid

◆ p_bs_grid

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
std::array<const void*, NumBTensor> ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::p_bs_grid

◆ p_ds_grid

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
std::array<const void*, NumDTensor> ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::p_ds_grid

◆ p_e_grid

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
void* ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::p_e_grid

◆ StrideAs

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
std::array<index_t, NumATensor> ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::StrideAs

◆ StrideBs

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
std::array<index_t, NumBTensor> ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::StrideBs

◆ StrideDs

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
std::array<index_t, NumDTensor> ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::StrideDs

◆ StrideE

template<index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0>
index_t ck::tensor_operation::device::GroupedGemmMultiABDKernelArgument< NumATensor, NumBTensor, NumDTensor >::StrideE
The documentation for this struct was generated from the following file: