#6738 [rollout] fix: skip redundant clone in get_named_tensor_buckets to avoid OOM during SGLang weight sync
原始 PR · 作者 HaozheZhang6 · 合并时间 2026-06-16 09:25
修复 SGLang 权重同步时冗余 clone 导致的 OOM
值得精读。这是一个典型的“一行改动解决大问题”的 PR,展示了如何通过理解张量存储布局来避免冗余分配。`_compact_for_bucket` 函数可作为工具函数在类似场景复用。建议合并并回测 Megatron 路径(作者提到无法测试),确保不引入回归。
参与讨论