[FSDP][optim_state_dict] Add device to _shard_utils.py to explicitly use the device from fsdp_state #109631
+28
−9
Facebook GitHub Tools / Meta Internal-Only Changes Check
succeeded
Sep 19, 2023 in 0s
This Pull Request and its Phabricator Diff are identical
Loading