pytorch/test/distributed/_composable
2024-10-25 17:19:34 +00:00
..
fsdp [compiled autograd] tls access helpers (#138061) 2024-10-22 08:03:52 +00:00
fully_shard [FSDP2] support torch._foreach_copy_(float8) for fully_shard(Float8Linear) (#135955) 2024-10-07 16:36:31 +00:00
test_composability [Pipelining] Refactor Interleaved1F1B and ZeroBubble (#137783) 2024-10-16 03:05:14 +00:00
test_checkpoint.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_compose.py
test_contract.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_replicate.py [DDP][FSDP2] keep DTensor params for replicate(fully_shard) (#133059) 2024-08-09 18:38:05 +00:00
test_replicate_with_compiler.py [inductor][memory] restructuring memory.py and turn on the flag (#137205) 2024-10-25 17:19:34 +00:00