#57112 [ROCm][CI] Fix MLA RoPE fused-kernel tests for TheRock image
原始 PR · 作者 mawong-amd · 合并时间 2026-09-16 13:57
ROCm MLA RoPE 测试改为按 ULP 分级容差
值得精读,尤其是从事 ROCm 内核、数值测试或 CI 稳定性的工程师:这是“不要靠调 atol/rtol 掩盖数值差异”的一个正面范例,用 FMA 预舍入选择推导出 2 product-ULP 的物理上界,再用失败比例上限约束整体行为。阅读顺序建议:先看 PR body 中 Triton 3.7 vs 3.8 的 lowering 对比理解动机,再看 `assert_rope_close_rocm` 的 Tier 1/Tier 2 双段断言与 NeoX/交错两套乘积推导,最后看 `test_concat_and_cache_mla_rope_fused` 如何用 slot_mapping 把 KV cache 的 rope 区段收集回 token 顺序、并把 nope 区段保持严格比对。需要留意的设计决策是 bf16 的 `max_fail_fraction=1e-2` 取值依据(仅由 eps 更粗解释)以及 fp8 分支未被纳入统一模型;若后续要推广该断言,应先确认这两个边界。