#27053 [BCG][GLM5] perf: BCG support and prefill enhancements
原始 PR · 作者 mattteochen · 合并时间 2026-06-25 04:13
为GLM5引入BCG支持并优化prefill性能
强烈建议精读。本 PR 展示了如何在推理框架中平衡 eager 与 CUDA 图执行,是 CUDA 图后端演进的关键步骤。特别关注 `MlaBmmFusionPlan` 和 `mla_bmm_then_unified_attention` 的设计,以及在 split op 中灵活切换 eager 与图的策略。讨论中的设计取舍值得团队参考。
参与讨论