Michael Voznesensky 42660015b4 [Dynamo x FSDP][2/x] Small changes to distributed to make it dynamo friendly (#106886) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/106886 Approved by: https://github.com/awgu, https://github.com/wconstab ghstack dependencies: #106884		2023-08-11 22:35:50 +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	[Dynamo x FSDP][2/x] Small changes to distributed to make it dynamo friendly (#106886)	2023-08-11 22:35:50 +00:00
fully_shard.py	[FSDP][9/N] Introduce CustomPolicy (#104986)	2023-08-03 12:46:36 +00:00
replicate.py	[Replicate] Add unit test with replicate param names (#102401)	2023-05-31 18:41:03 +00:00