pytorch/test/distributed
2024-06-20 19:23:32 +00:00
..
_composable [Horace's PR #126446] Prevent partitioner from ever saving views (#129039) 2024-06-19 23:21:16 +00:00
_shard
_spmd [DeviceMesh] Update get_group and add get_all_groups (#128097) 2024-06-08 04:28:56 +00:00
_tensor [BE] Skip the test if CUDA is not available (#128885) 2024-06-18 07:02:44 +00:00
_tools Extended Module Tracker (#128508) 2024-06-14 19:48:46 +00:00
algorithms
bin
checkpoint Pass device to is_pinned call inside TensorProperties.create_from_tensor (#128896) 2024-06-19 08:50:46 +00:00
elastic Pass params to dump_nccl_trace_pickle (#128781) 2024-06-18 03:46:57 +00:00
fsdp
launcher Validate logs are created by default (#128522) 2024-06-17 20:07:13 +00:00
nn/jit
optim
pipelining [pipelining] Rename to stage.py and schedules.py (#128278) 2024-06-08 04:42:35 +00:00
rpc
tensor/parallel [tp] refactor and fix PrepareModuleInput for DTensor inputs (#128431) 2024-06-12 19:16:33 +00:00
argparse_util_test.py
test_c10d_common.py
test_c10d_functional_native.py
test_c10d_gloo.py
test_c10d_logger.py
test_c10d_nccl.py [Test] Add test for only_active flag (#128191) 2024-06-11 22:26:01 +00:00
test_c10d_object_collectives.py
test_c10d_ops_nccl.py
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
test_control_collectives.py [c10d] add a simple test to demonstrate the user usage of collectives (#127665) 2024-06-05 04:32:11 +00:00
test_cuda_p2p.py force_stride_order on fused_all_gather_matmul/fused_matmul_reduce_scatter's operands to avoid a copy due to layout transformation (#127454) 2024-06-13 17:52:37 +00:00
test_data_parallel.py
test_device_mesh.py [DeviceMesh] Update get_group and add get_all_groups (#128097) 2024-06-08 04:28:56 +00:00
test_distributed_spawn.py
test_dynamo_distributed.py [dynamo] Remove ID_MATCH for FSDPModuleVariable (#129015) 2024-06-20 19:23:32 +00:00
test_fake_pg.py
test_functional_api.py
test_inductor_collectives.py Remove compile_threads=1 in test_inductor_collectives.py (#128580) 2024-06-18 19:31:13 +00:00
test_launcher.py
test_multi_threaded_pg.py
test_nccl.py
test_pg_wrapper.py
test_store.py [c10d] Socket, TCPStore: add better logging (#128673) 2024-06-14 23:08:29 +00:00
test_symmetric_memory.py Introduce a prototype for SymmetricMemory (#128582) 2024-06-19 03:38:58 +00:00