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
fully_shard	[BE]: FURB142 - Remove set mutations. Use set update (#124551)	2024-04-21 14:12:33 +00:00
test_checkpoint.py
test_compose.py
test_contract.py
test_replicate.py
test_replicate_with_compiler.py	[DDP][PT2D] Fix no_compiled_forward flag in the test (#124829)	2024-04-25 04:55:39 +00:00