pytorch/test/distributed/_composable
Andrew Gu 20eaa49dde [PT-D] Made _get_registry return None if no APIs applied (#113654)
I prefer to not modify the module if it does not have any of our APIs applied. The side effect of inserting a registry on the module when calling a getter is non-intuitive to me.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/113654
Approved by: https://github.com/fegin
2023-11-14 20:28:11 +00:00
..
fully_shard [state_dict] Move _gather_state_dict to dcp module (#112835) 2023-11-08 19:42:56 +00:00
test_checkpoint.py
test_compose.py
test_contract.py [PT-D] Made _get_registry return None if no APIs applied (#113654) 2023-11-14 20:28:11 +00:00
test_replicate.py