FusedMoeGemmHostArgs Struct Reference

FusedMoeGemmHostArgs Struct Reference#

Composable Kernel: ck_tile::FusedMoeGemmHostArgs Struct Reference

#include <fused_moegemm_kernel.hpp>

Public Attributes
const void *	a_ptr
const void *	a_scale_ptr
const void *	g_ptr
const void *	d_ptr
const void *	g_scale_ptr
const void *	d_scale_ptr
const void *	y_smooth_scale_ptr
void *	o_ptr
const void *	sorted_token_ids_ptr
const void *	sorted_weight_ptr
const void *	sorted_expert_ids_ptr
const void *	num_sorted_tiles_ptr
index_t	hidden_size
index_t	intermediate_size
index_t	num_tokens
index_t	num_experts
index_t	topk
index_t	stride_token

Member Data Documentation

◆ a_ptr

const void* ck_tile::FusedMoeGemmHostArgs::a_ptr

◆ a_scale_ptr

const void* ck_tile::FusedMoeGemmHostArgs::a_scale_ptr

◆ d_ptr

const void* ck_tile::FusedMoeGemmHostArgs::d_ptr

◆ d_scale_ptr

const void* ck_tile::FusedMoeGemmHostArgs::d_scale_ptr

◆ g_ptr

const void* ck_tile::FusedMoeGemmHostArgs::g_ptr

◆ g_scale_ptr

const void* ck_tile::FusedMoeGemmHostArgs::g_scale_ptr

◆ hidden_size

index_t ck_tile::FusedMoeGemmHostArgs::hidden_size

◆ intermediate_size

index_t ck_tile::FusedMoeGemmHostArgs::intermediate_size

◆ num_experts

index_t ck_tile::FusedMoeGemmHostArgs::num_experts

◆ num_sorted_tiles_ptr

const void* ck_tile::FusedMoeGemmHostArgs::num_sorted_tiles_ptr

◆ num_tokens

index_t ck_tile::FusedMoeGemmHostArgs::num_tokens

◆ o_ptr

void* ck_tile::FusedMoeGemmHostArgs::o_ptr

◆ sorted_expert_ids_ptr

const void* ck_tile::FusedMoeGemmHostArgs::sorted_expert_ids_ptr

◆ sorted_token_ids_ptr

const void* ck_tile::FusedMoeGemmHostArgs::sorted_token_ids_ptr

◆ sorted_weight_ptr

const void* ck_tile::FusedMoeGemmHostArgs::sorted_weight_ptr

◆ stride_token

index_t ck_tile::FusedMoeGemmHostArgs::stride_token

◆ topk

index_t ck_tile::FusedMoeGemmHostArgs::topk

◆ y_smooth_scale_ptr

const void* ck_tile::FusedMoeGemmHostArgs::y_smooth_scale_ptr

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

fused_moegemm_kernel.hpp