pytorch

1439cb03 - [FSDP][optim_state_dict][9/N] Rewrite the all-gather flow of optimizer state to support older GPUs (#91343)

[FSDP][optim_state_dict][9/N] Rewrite the all-gather flow of optimizer state to support older GPUs (#91343) Pull Request resolved: https://github.com/pytorch/pytorch/pull/91343 Approved by: https://github.com/rohan-varma