pytorch/test/distributed
Ke Wen 062387fb53 [SymmMem] Speed up tests (#153677)
Use `MultiProcContinousTest` to avoid re-create ProcessGroup in each test instance.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153677
Approved by: https://github.com/fegin, https://github.com/Skylion007, https://github.com/ngimel
ghstack dependencies: #153653
2025-05-26 03:39:11 +00:00
..
_composable [Graph Partition] support removed arguments, NoneLayout, and mutation (#153899) 2025-05-22 04:24:31 +00:00
_shard add support for 0 size shardedTensor and recalculate metadata from all_gather (#152583) 2025-05-03 17:26:29 +00:00
_tools [BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225) 2025-05-09 20:40:54 +00:00
algorithms
bin
checkpoint [DSD] Don't pop tensors if they are on Meta device (#153185) 2025-05-16 07:18:39 +00:00
elastic elastic: do not shutdown rendezvous on leaving workers (#152525) 2025-05-14 00:44:10 +00:00
flight_recorder [c10d][fr] Fix another bug when we should continue when the op list is empty (#151798) 2025-04-22 04:43:31 +00:00
fsdp [BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225) 2025-05-09 20:40:54 +00:00
launcher
nn/jit
optim
pipelining [Distributed][CI] Rework continuous TestCase (#153653) 2025-05-25 03:49:29 +00:00
rpc
tensor [DTensor] enable SimpleFSDP's composability with Tensor Parallel (#152286) 2025-05-23 01:40:38 +00:00
_test_template.py [Distributed][CI] Rework continuous TestCase (#153653) 2025-05-25 03:49:29 +00:00
argparse_util_test.py
test_backends.py
test_c10d_common.py [c10d] Allow split_group to work with non nccl backends (#152175) 2025-05-16 00:15:29 +00:00
test_c10d_functional_native.py Add a test for AsyncCollectiveTensor handling for maybe-view ops (#152688) 2025-05-05 17:21:00 +00:00
test_c10d_gloo.py Fix tcp init when using port 0 (#154156) 2025-05-23 21:41:58 +00:00
test_c10d_logger.py
test_c10d_nccl.py [c10d] Add support for testing SIGABRT return (#153167) 2025-05-26 00:56:05 +00:00
test_c10d_object_collectives.py
test_c10d_ops_nccl.py [Distributed][CI] Rework continuous TestCase (#153653) 2025-05-25 03:49:29 +00:00
test_c10d_pypg.py c10d/ProcessGroup: cleanup abort and shutdown (#148798) 2025-03-08 18:33:18 +00:00
test_c10d_spawn_gloo.py
test_c10d_spawn_nccl.py
test_c10d_spawn_ucc.py
test_c10d_spawn.py
test_c10d_ucc.py
test_collective_utils.py
test_composability.py [Distributed][CI] Rework continuous TestCase (#153653) 2025-05-25 03:49:29 +00:00
test_compute_comm_reordering.py [inductor][comms] skip reorder_for_locality for wait nodes (#150074) 2025-04-16 10:18:33 +00:00
test_control_collectives.py
test_data_parallel.py
test_device_mesh.py [device_mesh] improve device selection logic (#150897) 2025-05-14 06:29:16 +00:00
test_distributed_spawn.py
test_dynamo_distributed.py [dynamo] unimplemented -> unimplemented_v2 in variables/builtin.py (#151145) 2025-04-16 17:16:05 +00:00
test_fake_pg.py [BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225) 2025-05-09 20:40:54 +00:00
test_functional_api.py Revert "[inductor][dynamo] Include operator name in size/stride/alignment assertion (#152353)" 2025-05-16 08:20:39 +00:00
test_inductor_collectives.py [Graph Partition] Maintain relative order within partition during reordering (#153111) 2025-05-09 18:49:53 +00:00
test_launcher.py
test_multi_threaded_pg.py
test_nccl.py
test_nvshmem.py [Distributed][CI] Rework continuous TestCase (#153653) 2025-05-25 03:49:29 +00:00
test_pg_wrapper.py
test_serialization.py
test_store.py c10d/Store: add nonblocking mode to queue_pop (#151485) 2025-04-18 02:14:50 +00:00
test_symmetric_memory.py [SymmMem] Speed up tests (#153677) 2025-05-26 03:39:11 +00:00