/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fused_moe/pipeline/fused_moegemm_traits.hpp File Reference

• develop
• include
• ck_tile
• ops
• fused_moe
• pipeline

Classes | Namespaces | Enumerations

fused_moegemm_traits.hpp File Reference

#include "ck_tile/core.hpp"

Go to the source code of this file.

Classes
struct	ck_tile::FusedMoeGemmTraits< IsGateOnly_, UseSmoothQuant_, OAtomic_, PermuteEnum_, PadHiddenSize_, PadIntermediateSize_, PipeInterleave_ >

Namespaces
	ck_tile

Enumerations
enum class	ck_tile::FusedMoeGemmWeightPermuteEnum {   ck_tile::no_permute = 0 ,   ck_tile::b_nr_kr_kw_nw_kv = 1 ,   ck_tile::b_nr_kr_waveflatten = b_nr_kr_kw_nw_kv }
enum class	ck_tile::FusedMoeGemmPipelineSequencerEnum {   ck_tile::SLD_A = 1 << 0 ,   ck_tile::SLD_B = 1 << 1 ,   ck_tile::GLD_A = 1 << 2 ,   ck_tile::GLD_B = 1 << 3 ,   ck_tile::SST_A = 1 << 4 ,   ck_tile::SST_B = 1 << 5 ,   ck_tile::GST_O = 1 << 6 }