#41882 Add NVFP4 all-gather GEMM fusion for AsyncTP
原始 PR · 作者 baonudesifeizhai · 合并时间 2026-05-10 09:13
为 AsyncTP 添加 NVFP4 all-gather GEMM 融合路径
推荐精读,尤其关注 `collective_fusion.py` 中 `FlashInferAllGatherFP4Pattern` 的 `pattern` 与 `replacement` 设计,以及 `sequence_parallelism.py` 中 NVFP4 量化与序列平行的整合方式。对推理性能优化感兴趣的同学可以关注 reduce-scatter 融合的后续进展。
参与讨论