Kargs Struct Reference

Kargs Struct Reference#

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

#include <moe_sorting_kernel.hpp>

Public Attributes

const void * p_topk_ids
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_sorted_expert_ids
void * p_total_tokens_post_pad
void * p_moe_buf
index_t tokens
index_t num_experts
long_index_t moe_buf_bytes
index_t tokens_per_thread
index_t smem_rows
mdiv unit_size_mdiv
mdiv topk_mdiv
mdiv expert_mdiv

Member Data Documentation

◆ expert_mdiv

template<typename Problem_>
mdiv ck_tile::MoeSortingKernel< Problem_ >::Kargs::expert_mdiv

◆ moe_buf_bytes

template<typename Problem_>
long_index_t ck_tile::MoeSortingKernel< Problem_ >::Kargs::moe_buf_bytes

◆ num_experts

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

◆ p_local_expert_mask

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

◆ p_local_tokens

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

◆ p_moe_buf

template<typename Problem_>
void* ck_tile::MoeSortingKernel< Problem_ >::Kargs::p_moe_buf

◆ p_sorted_expert_ids

template<typename Problem_>
void* ck_tile::MoeSortingKernel< Problem_ >::Kargs::p_sorted_expert_ids

◆ p_sorted_token_ids

template<typename Problem_>
void* ck_tile::MoeSortingKernel< Problem_ >::Kargs::p_sorted_token_ids

◆ p_sorted_weights

template<typename Problem_>
void* ck_tile::MoeSortingKernel< Problem_ >::Kargs::p_sorted_weights

◆ p_topk_ids

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

◆ p_total_tokens_post_pad

template<typename Problem_>
void* ck_tile::MoeSortingKernel< Problem_ >::Kargs::p_total_tokens_post_pad

◆ p_weights

template<typename Problem_>
const void* ck_tile::MoeSortingKernel< Problem_ >::Kargs::p_weights

◆ smem_rows

template<typename Problem_>
index_t ck_tile::MoeSortingKernel< Problem_ >::Kargs::smem_rows

◆ tokens

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

◆ tokens_per_thread

template<typename Problem_>
index_t ck_tile::MoeSortingKernel< Problem_ >::Kargs::tokens_per_thread

◆ topk_mdiv

template<typename Problem_>
mdiv ck_tile::MoeSortingKernel< Problem_ >::Kargs::topk_mdiv

◆ unit_size_mdiv

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