#52737 [ROCm][Perf] Fuse DeepSeek-V4 mHC post/pre and RMSNorm with AITER
原始 PR · 作者 shen-shanshan · 合并时间 2026-08-20 12:51
ROCm 融合 DSV4 mHC 与 RMSNorm 内核,吞吐约增 1%
值得精读,尤其是对 kernel 融合接入模式和 ROCm 后端开发有兴趣的工程师。重点关注三点:`_aiter_ops.py` 中 `mhc_fused_post_pre` 的返回值顺序转换与空 token 分支、`model.py` 中融合能力的条件探测(`hc_mult == 4` + hidden size 白名单)、以及 `mhc.py` 各 `forward_hip` 中 AITER/TileLang/torch 的三级回退结构。若后续推广类似融合,建议补上覆盖 fake/meta 与回退路径的单元测试,并考虑把 hidden size 白名单收敛为 AITER 侧的查询接口,避免多处硬编码。