pytorch

66d07e3b - [FSDP] Only move current FSDP's states to GPU during init (#98319)

[FSDP] Only move current FSDP's states to GPU during init (#98319) Fixes https://github.com/pytorch/pytorch/issues/95813 Pull Request resolved: https://github.com/pytorch/pytorch/pull/98319 Approved by: https://github.com/rohan-varma