#25920 [bugfix] Honor cast_x_before_out_mul in RMSNorm.forward_cuda residual path
原始 PR · 作者 charlotte12l · 合并时间 2026-05-28 16:22
修复RMSNorm残差路径忽略cast_x_before_out_mul标志
这是一个高质量 bugfix,修复了影响核心正确性的问题,且设计迭代清晰——从临时 fallback 到独立 kernel 再到合并到现有 kernel。值得精读:展示了如何在 CUDA kernel 中通过 `if constexpr` 实现多语义路径,以及如何平衡数值精度与性能。建议相关模型维护者关注黄金测试是否需要调整。
参与讨论