pytorch

f5a0b31a - [FSDP][optim_state_dict] Make FSDP optim_state_dict aware of DDP prefix (#96415)

[FSDP][optim_state_dict] Make FSDP optim_state_dict aware of DDP prefix (#96415) Summary: When wrapping FSDP within DDP, optimizer state_dict may be broken due to the prefix of DDP. This PR fixes the issue. Test Plan: CI Differential Revision: D43893609 Pull Request resolved: https://github.com/pytorch/pytorch/pull/96415 Approved by: https://github.com/zhaojuanmao