pytorch
d54e20f4 - [FSDP][state_dict] Add a unittest for local_state_dict resharding (#110625)

Commit
1 year ago
[FSDP][state_dict] Add a unittest for local_state_dict resharding (#110625) This PR adds a unittest to demonstrate the ability for LOCAL_STATE_DICT to do resharding. Differential Revision: [D44260141](https://our.internmc.facebook.com/intern/diff/D44260141/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/110625 Approved by: https://github.com/wz337
Author
Committer
Parents
Loading