pytorch/test/distributed
2023-06-22 00:36:35 +00:00
..
_composable [dynamo x fsdp] Simplify stream logic handling (#103902) 2023-06-21 01:34:19 +00:00
_shard [distributed][sharded_tensor] Move local_shards check from ShardedTensorBase to ShardedTensor (#100197) 2023-05-02 12:42:24 +00:00
_spmd [distributed] implemented find_all_descendants (#102138) 2023-05-24 21:47:59 +00:00
_tensor [dtensor] support torch.save/load with DTensor (#103106) 2023-06-09 04:11:15 +00:00
_tools
algorithms [BE] Fix all B022 useless-contextlib-suppress (#100335) 2023-04-30 18:47:40 +00:00
bin
checkpoint [DCP] Rewrite read slicing to use a wrapper. (#99167) 2023-06-08 13:52:13 +00:00
elastic [RFC] Allow elastic agent to fail fast (#99051) 2023-04-25 23:51:20 +00:00
fsdp [FSDP][4/n]Update use_dtensor option for _optim_utils.py (#103599) 2023-06-14 20:18:33 +00:00
launcher Fix "sandcastle_skip_if decorator name is confusing" (#95649) 2023-03-03 09:29:40 +00:00
nn/jit
optim [BE] Fix all B022 useless-contextlib-suppress (#100335) 2023-04-30 18:47:40 +00:00
pipeline/sync Run tests in USE_PYTEST_LIST through run_tests (#95659) 2023-02-28 22:09:01 +00:00
rpc Revert "Revert "Expandable blocks in allocator (#96995)"" (#99275) 2023-04-17 23:46:08 +00:00
tensor/parallel [TP] Enable more generic attn in Tensor Parallelism (#100508) 2023-05-07 18:15:49 +00:00
argparse_util_test.py
test_c10d_common.py Allow ORT backend for DTensor (#101914) 2023-06-01 22:37:09 +00:00
test_c10d_gloo.py Enable test sparse allreduce basics Windows (#103317) 2023-06-14 07:37:50 +00:00
test_c10d_logger.py [c10d] Record time spent for init_process_group, new_group, _store_based_barrier (#101912) 2023-05-24 09:36:34 +00:00
test_c10d_nccl.py NCCL process group: avoid workEnqueue when capturing cuda graph (#103503) 2023-06-13 23:12:43 +00:00
test_c10d_object_collectives.py [c10d] Remove test for init barrier (#103223) 2023-06-08 16:56:40 +00:00
test_c10d_pypg.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_c10d_spawn.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_c10d_spawn_gloo.py Fix "sandcastle_skip_if decorator name is confusing" (#95649) 2023-03-03 09:29:40 +00:00
test_c10d_spawn_nccl.py Fix "sandcastle_skip_if decorator name is confusing" (#95649) 2023-03-03 09:29:40 +00:00
test_c10d_spawn_ucc.py Fix "sandcastle_skip_if decorator name is confusing" (#95649) 2023-03-03 09:29:40 +00:00
test_c10d_ucc.py [CI] Enable UCC in CI (#100395) 2023-06-08 19:01:22 +00:00
test_data_parallel.py Fix "sandcastle_skip_if decorator name is confusing" (#95649) 2023-03-03 09:29:40 +00:00
test_distributed_spawn.py Revert "[DDP] multiple forward support for static graph (#103487)" (#103873) 2023-06-20 16:25:00 +00:00
test_dynamo_distributed.py Enable additional inductor test suites on ROCm (#102270) 2023-06-22 00:36:35 +00:00
test_fake_pg.py [c10d] add fake pg necessary collectives (#102238) 2023-05-25 05:01:16 +00:00
test_functional_api.py Implement all_gather_into_tensor_coalesced. (#98642) 2023-06-13 15:06:52 +00:00
test_inductor_collectives.py Enable additional inductor test suites on ROCm (#102270) 2023-06-22 00:36:35 +00:00
test_launcher.py [BE] Prefer dash over underscore in command-line options (#94505) 2023-02-09 20:16:49 +00:00
test_multi_threaded_pg.py [Oncall][MTPG] Fix flaky test multi_threaded - test_broadcast_object_list (#103568) 2023-06-18 07:05:28 +00:00
test_nccl.py Reduce pytest blocklist (#96016) 2023-03-07 18:30:27 +00:00
test_pg_wrapper.py [c10d] Figure out device to use for object collectives (#100954) 2023-05-11 01:49:09 +00:00
test_store.py [gloo] Make it possible for gloo TCPStore to take over an existing socket fd (#103478) 2023-06-16 17:15:56 +00:00