Wanchao Liang c59a2369be [fsdp2] Accomodate FSDP2 to accept parent mesh > 2 (#125778) ... as titled, to support higher dimension parallelism Pull Request resolved: https://github.com/pytorch/pytorch/pull/125778 Approved by: https://github.com/weifengpy		2024-05-09 05:02:21 +00:00
..
fsdp	[fsdp2] Accomodate FSDP2 to accept parent mesh > 2 (#125778)	2024-05-09 05:02:21 +00:00
__init__.py
checkpoint_activation.py
contract.py
fully_shard.py	[Composable APIs] Add composable API fully_shard deprecation warning (#120929)	2024-03-01 00:55:16 +00:00
replicate.py	[PT2D] Ensure the trace rules are correct with distributed (#125333)	2024-05-02 16:28:38 +00:00