Kargs Struct Reference

Kargs Struct Reference#

Composable Kernel: ck_tile::MoeSortingMultiPhaseKernel_P0_v1< Problem_ >::Kargs Struct Reference
Public Attributes | List of all members
ck_tile::MoeSortingMultiPhaseKernel_P0_v1< Problem_ >::Kargs Struct Reference

#include <moe_sorting_kernel.hpp>

Public Attributes

const void * p_topk_ids
const void * p_local_tokens
void * p_expert_mesh
index_t tokens
index_t num_experts
index_t mesh_stride
mdiv topk_mdiv

Member Data Documentation

◆ mesh_stride

template<typename Problem_>
index_t ck_tile::MoeSortingMultiPhaseKernel_P0_v1< Problem_ >::Kargs::mesh_stride

◆ num_experts

template<typename Problem_>
index_t ck_tile::MoeSortingMultiPhaseKernel_P0_v1< Problem_ >::Kargs::num_experts

◆ p_expert_mesh

template<typename Problem_>
void* ck_tile::MoeSortingMultiPhaseKernel_P0_v1< Problem_ >::Kargs::p_expert_mesh

◆ p_local_tokens

template<typename Problem_>
const void* ck_tile::MoeSortingMultiPhaseKernel_P0_v1< Problem_ >::Kargs::p_local_tokens

◆ p_topk_ids

template<typename Problem_>
const void* ck_tile::MoeSortingMultiPhaseKernel_P0_v1< Problem_ >::Kargs::p_topk_ids

◆ tokens

template<typename Problem_>
index_t ck_tile::MoeSortingMultiPhaseKernel_P0_v1< Problem_ >::Kargs::tokens

◆ topk_mdiv

template<typename Problem_>
mdiv ck_tile::MoeSortingMultiPhaseKernel_P0_v1< Problem_ >::Kargs::topk_mdiv
The documentation for this struct was generated from the following file: