#21321 [Kernel] Support FlashInfer TRTLLM-Gen fused MoE for non-gated FP4 & FP8 (Nemotron)
原始 PR · 作者 danielafrimi · 合并时间 2026-04-29 16:28
FlashInfer TRTLLM-Gen 融合 MoE 支持非门控 FP4/FP8,加速 NemotronH-120B
值得精读。该 PR 清晰展示了如何为 FlashInfer TRTLLM-Gen MoE 后端扩展非门控激活支持,包括权重对齐策略、激活类型传递和自动 backend 选择。设计中的分支权衡和测试取舍也值得关注。建议重点关注 `_align_fp8_moe_weights` 函数的对齐逻辑和 `activation_type` 参数传递链。
参与讨论