pytorch

01fdcbdc - [FSDP][optim_state_dict][Easy] Temporarily disable rank0_only=True for use_orig_paramscaseEas (#99354)

[FSDP][optim_state_dict][Easy] Temporarily disable rank0_only=True for use_orig_paramscaseEas (#99354) Summary: We have not made use_orig_params=True support rank0_only optimizer_state_dict. Test Plan: CI Reviewed By: wz337 Differential Revision: D45054041 Pull Request resolved: https://github.com/pytorch/pytorch/pull/99354 Approved by: https://github.com/wz337