Kargs Struct Reference

• ck_tile
• MoeSortingMultiPhaseKernel_P23
• Kargs

Public Attributes | List of all members

ck_tile::MoeSortingMultiPhaseKernel_P23< 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_expert_mesh
void *	p_expert_cumsum
void *	p_total_tokens_post_pad
void *	p_sorted_expert_ids
void *	p_sorted_token_ids
void *	p_sorted_weights
void *	p_moe_buf
index_t	tokens
index_t	num_experts
index_t	mesh_stride
mdiv	unit_size_mdiv
mdiv	topk_mdiv
long_index_t	moe_buf_bytes

Member Data Documentation

◆ mesh_stride

template<typename Problem_>

index_t ck_tile::MoeSortingMultiPhaseKernel_P23< Problem_ >::Kargs::mesh_stride

◆ moe_buf_bytes

template<typename Problem_>

long_index_t ck_tile::MoeSortingMultiPhaseKernel_P23< Problem_ >::Kargs::moe_buf_bytes

◆ num_experts

template<typename Problem_>

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

◆ p_expert_cumsum

template<typename Problem_>

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

◆ p_expert_mesh

template<typename Problem_>

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

◆ p_local_expert_mask

template<typename Problem_>

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

◆ p_local_tokens

template<typename Problem_>

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

◆ p_moe_buf

template<typename Problem_>

void* ck_tile::MoeSortingMultiPhaseKernel_P23< Problem_ >::Kargs::p_moe_buf

◆ p_sorted_expert_ids

template<typename Problem_>

void* ck_tile::MoeSortingMultiPhaseKernel_P23< Problem_ >::Kargs::p_sorted_expert_ids

◆ p_sorted_token_ids

template<typename Problem_>

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

◆ p_sorted_weights

template<typename Problem_>

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

◆ p_total_tokens_post_pad

template<typename Problem_>

void* ck_tile::MoeSortingMultiPhaseKernel_P23< Problem_ >::Kargs::p_total_tokens_post_pad

◆ p_weights

template<typename Problem_>

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

◆ tokens

template<typename Problem_>

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

◆ topk_mdiv

template<typename Problem_>

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

◆ unit_size_mdiv

template<typename Problem_>

mdiv ck_tile::MoeSortingMultiPhaseKernel_P23< Problem_ >::Kargs::unit_size_mdiv

---

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

• moe_sorting_kernel.hpp