#24459 Register aten::rms_norm and aten::mm.dtype in batch invariant mode
原始 PR · 作者 merrymercy · 合并时间 2026-05-06 08:21
功能
重要性 5.67
洞察度 4.00
在 batch invariant 模式中注册 rms_norm 和 mm.dtype
该 PR 逻辑清晰、改动精简,值得关注的是其对 batch-invariant 兼容层的扩展模式,展示了如何为更多 ATen 算子添加确定性支持。建议后续测试覆盖新增算子的确定性行为。