pytorch
3d1c5c89 - [FSDP][state_dict][4/N] Move the core logic of summon full parameters to _unshard_params_utils.py (#88636)

Commit
2 years ago
[FSDP][state_dict][4/N] Move the core logic of summon full parameters to _unshard_params_utils.py (#88636) **What** `_summon_full_parameters` is required for state_dict. To enable composable FSDP state_dict, `_summon_full_params` must be accessible without FullyShardedDataParall. This PR move the core logic of `_summon_full_params` to `_unshard_params_utils`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/88636 Approved by: https://github.com/awgu
Author
Committer
Parents
Loading