pytorch/test/distributed
2024-10-25 17:35:27 +00:00
..
_composable [inductor][memory] restructuring memory.py and turn on the flag (#137205) 2024-10-25 17:19:34 +00:00
_shard
_tensor Fix unbind_copy and add its decomposition (#134319) 2024-10-23 19:13:44 +00:00
_tools Upgrade distributed test to g4dn instances (T4 GPUs) (#137161) 2024-10-20 23:48:54 +00:00
algorithms
bin
checkpoint [Distributed][Test] Fix todo in distributed test files (#136836) 2024-10-16 01:15:12 +00:00
elastic
flight_recorder
fsdp [Distributed][Test] Fix todo in distributed test files (#136836) 2024-10-16 01:15:12 +00:00
launcher
nn/jit
optim
pipelining [Pipelining] small comments and variable renames (#138735) 2024-10-25 12:06:54 +00:00
rpc
tensor/parallel Allow parallelize_module to get device_mesh from ambient context (#134247) 2024-10-09 00:19:03 +00:00
argparse_util_test.py
test_c10d_common.py
test_c10d_functional_native.py Revert "[c10d][Partial-Graph Overlap] Support calling .wait_tensor() within compiled region on output tensor of eager async_op=True collective (#137763)" 2024-10-24 17:46:09 +00:00
test_c10d_gloo.py
test_c10d_logger.py
test_c10d_nccl.py [c10d] allow sub group to be eagerly inited even if default one is not (#138665) 2024-10-24 23:51:28 +00:00
test_c10d_object_collectives.py Revert "[c10d][CI] Improve world size setting in some tests (#138846)" 2024-10-25 17:35:27 +00:00
test_c10d_ops_nccl.py Revert "[c10d][CI] Improve world size setting in some tests (#138846)" 2024-10-25 17:35:27 +00:00
test_c10d_pypg.py
test_c10d_spawn.py
test_c10d_spawn_gloo.py
test_c10d_spawn_nccl.py
test_c10d_spawn_ucc.py
test_c10d_ucc.py
test_collective_utils.py
test_compute_comm_reordering.py [CI] Add Compiled DDP / Compiled FSDP2 / compute-comm reordering tests to test_inductor_distributed (#138178) 2024-10-20 19:38:18 +00:00
test_control_collectives.py
test_data_parallel.py
test_device_mesh.py [DeviceMesh] Use split_group to create sub_groups for nccl backend if the default pg is eagerly initialized (#138129) 2024-10-22 00:00:05 +00:00
test_distributed_spawn.py
test_dynamo_distributed.py
test_fake_pg.py
test_functional_api.py
test_inductor_collectives.py Revert "[c10d][Partial-Graph Overlap] Support calling .wait_tensor() within compiled region on output tensor of eager async_op=True collective (#137763)" 2024-10-24 17:46:09 +00:00
test_launcher.py
test_multi_threaded_pg.py
test_nccl.py
test_pg_wrapper.py
test_store.py [TCPStore] Throw value error if passing world_size=0 to TCPStore (#137792) 2024-10-11 23:42:57 +00:00
test_symmetric_memory.py [fused_all_gather_scaled_matmul] support rowwise scaling (#137805) 2024-10-15 21:35:14 +00:00