Skip to main content

Module moe

singe_kernel::cpu

Module moe 

Source
Expand description

Mixture-of-experts routing and alignment helpers.

Functionsยง

ceil_div
fused_moe_f8e4m3_block_scaled_f32
fused_moe_f32
moe_align_block_size_i32