#7610 [ckpt, fsdp] fix: FSDP model merger concatenates replicated buffers
原始 PR · 作者 Josephasafg · 合并时间 2026-08-31 14:45
修复 FSDP 合并器误拼接 replicated buffer 崩溃与 shape 损坏
值得精读。这是一个小而完整的数据契约修复:核心决策是识别 FSDP 下“非 DTensor 即完整复制”的语义,并用 fail-fast 替代静默损坏。可借鉴的点包括:错误信息携带 key 名与 rank 号、用 torch.equal 做全量一致性校验、配套聚焦的 CPU 单测直接锁定回归。