#21825 [ROCM][RL] Shuffle Weight In-Place to Preserve Parameter Attributes
原始 PR · 作者 zyzshishui · 合并时间 2026-04-03 14:43
修复ROCm/aiter后处理中权重替换丢失自定义属性问题,确保RL工作流正常。
建议精读unquant.py中的copy_or_rebind_param实现,理解其如何平衡原地更新与形状兼容;同时关注review中关于分片属性同步的讨论,这对分布式训练场景很重要。
参与讨论