pytorch

d991ce6d - [FSDP][3/N]_shard_utils update for dtensor state_dict support (#103479)

[FSDP][3/N]_shard_utils update for dtensor state_dict support (#103479) Same as https://github.com/pytorch/pytorch/pull/102545 (this branch is corrupted so have to re-submit). Pull Request resolved: https://github.com/pytorch/pytorch/pull/103479 Approved by: https://github.com/fegin