#28787 [AMD] Fix RMSNorm batch-invariance on ROCm under deterministic inference
原始 PR · 作者 yuankaichen-amd · 合并时间 2026-07-05 14:49
修复 ROCm RMSNorm 批次不变性
值得合并。这是一个针对 ROCm 平台的精确 bugfix,修复了确定性推理模式下 RMSNorm 的批次不变性问题,同时通过使用 Triton 核函数避免了性能回退。评审过程健康,反馈被及时采纳。无需精读。
参与讨论