Kargs Struct Reference

• ck_tile
• MoeSortingMultiPhaseKernel_P3
• Kargs

Public Attributes | List of all members

ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs Struct Reference

#include <moe_sorting_kernel.hpp>

Public Attributes
const void *	p_weights
const void *	p_local_expert_mask
const void *	p_local_tokens
void *	p_sorted_token_ids
void *	p_sorted_weights
void *	p_expert_mesh
void *	p_expert_cumsum
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_P3< Problem_ >::Kargs::mesh_stride

◆ num_experts

template<typename Problem_>

index_t ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs::num_experts

◆ p_expert_cumsum

template<typename Problem_>

void* ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs::p_expert_cumsum

◆ p_expert_mesh

template<typename Problem_>

void* ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs::p_expert_mesh

◆ p_local_expert_mask

template<typename Problem_>

const void* ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs::p_local_expert_mask

◆ p_local_tokens

template<typename Problem_>

const void* ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs::p_local_tokens

◆ p_sorted_token_ids

template<typename Problem_>

void* ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs::p_sorted_token_ids

◆ p_sorted_weights

template<typename Problem_>

void* ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs::p_sorted_weights

◆ p_weights

template<typename Problem_>

const void* ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs::p_weights

◆ tokens

template<typename Problem_>

index_t ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs::tokens

◆ topk_mdiv

template<typename Problem_>

mdiv ck_tile::MoeSortingMultiPhaseKernel_P3< Problem_ >::Kargs::topk_mdiv

---

The documentation for this struct was generated from the following file:

• moe_sorting_kernel.hpp