Skip to content

vllm.models.common.ops.moe_tail_fusion

Building blocks shared by the CuTe DSL kernels that fuse a MoE layer's tail (top-k finalize, TP all-reduce, norms) across models.

Modules:

  • primitives –

    CuTe DSL primitives shared by the fused MoE-tail kernels: PTX loads and