#6950 [trainer] fix: spmd_types and activation checkpointing composability bug
原始 PR · 作者 acisseJZhong · 合并时间 2026-07-07 10:52
缺陷修复
重要性 6.06
洞察度 5.00
修复 spmd_types 与选择性激活检查点兼容性崩溃
该 PR 值得阅读,特别是对 TorchTitan 引擎或分布式训练中线程本地 mesh 传播机制感兴趣的开发者。设计决策(禁用 autograd 多线程)参考了 TorchTitan 上游实践,思路清晰。注意兼容性风险,建议环境确认 PyTorch 版本 >=2.5.0。