pytorch

8f294f78 - [FSDP][optim_state_dict] Fix the conditions to check non-parameter associated states (#92744)

[FSDP][optim_state_dict] Fix the conditions to check non-parameter associated states (#92744) If a state is not associated with any parameter, `FSDP.optim_state_dict` should still save it. The current implementation to determine whether a state is associated with a parameter is not completely correct and can cause `use_orig_params=True` have extra states. Pull Request resolved: https://github.com/pytorch/pytorch/pull/92744 Approved by: https://github.com/awgu