Chien-Chin Huang b1fb591272 [replicate] Simplify replicate() init logic and remove unnecessary variables in _ReplicateState (#113679) ... Many variables _ReplicateState are created because replicate() was lazy initialized. This PR removes these variables and simplifes the logic.y Differential Revision: [D51317874](https://our.internmc.facebook.com/intern/diff/D51317874/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/113679 Approved by: https://github.com/awgu		2023-11-28 00:55:36 +00:00
..
__init__.py	Add global registry to composable API contract (#90579)	2022-12-10 22:41:10 +00:00
checkpoint_activation.py	[Composable] Use non-reentrant generator, remove reentrant (#105176)	2023-07-26 07:03:03 +00:00
contract.py	[PT-D] Made _get_registry return None if no APIs applied (#113654)	2023-11-14 20:28:11 +00:00
fully_shard.py	[FSDP][9/N] Introduce CustomPolicy (#104986)	2023-08-03 12:46:36 +00:00
replicate.py	[replicate] Simplify replicate() init logic and remove unnecessary variables in _ReplicateState (#113679)	2023-11-28 00:55:36 +00:00