Kargs Struct Reference

Kargs Struct Reference#

Composable Kernel: ck_tile::MoeSortingMultiPhaseKernel_P0_v2< Problem_ >::Kargs Struct Reference
Public Attributes | List of all members
ck_tile::MoeSortingMultiPhaseKernel_P0_v2< 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 mesh_stride
mdiv topk_mdiv
const void * p_local_expert_mask
void * p_expert_cumsum
index_t num_experts

Member Data Documentation

◆ mesh_stride

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

◆ num_experts

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

◆ p_expert_cumsum

template<typename Problem_>
void* ck_tile::MoeSortingMultiPhaseKernel_P0_v2< Problem_ >::Kargs::p_expert_cumsum

◆ p_expert_mesh

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

◆ p_local_expert_mask

template<typename Problem_>
const void* ck_tile::MoeSortingMultiPhaseKernel_P0_v2< Problem_ >::Kargs::p_local_expert_mask

◆ p_local_tokens

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

◆ p_topk_ids

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

◆ tokens

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

◆ topk_mdiv

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