trunk/841e3dc165963ca6550560eab70a84d1c166b9eb: [FSDP2] Fix resharding to a smaller mesh after forward (#200985)
- PyTorch 近 90 天出现 2413 次
- PyTorch 近 90 天第 2381 次发版
- 上一次：同一天稍早 · trunk/79ae8043a9e9029f67a3b2e7f7517aeb194c9f0e
发生了什么
BC-breaking? No: configurations that worked on main give the same results, and those that failed now work or raise a named NotImplementedError . Summary (human written only) Agent details (if applicable) AI-assisted implementation and summary prepared with Claude Code, split out of #200034 at the author's request. With reshard_after_forward=int , FSDP reshards parameters after forward to flat chunks of their unshard…
摘要按规则整理自下方来源原文