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