Skip to main content

moe_dp4a_kernels

Function moe_dp4a_kernels 

Source
pub fn moe_dp4a_kernels(two_pass: bool) -> String
Expand description

The sparse fused MoE kernel set for a coalesced int8 path: the precision-independent router + top-K pair, then the single-pass (default) or two-pass (logit-exact) quantize/gate-up-SwiGLU/down-combine trio. Public so a checked-in copy of the kernel set can be pinned byte-equal to what the compiler emits today.