pytorch/test/distributed
2023-10-11 02:37:55 +00:00
..
_composable Define the public API for torch.distributed.fsdp (#109922) 2023-09-28 02:15:58 +00:00
_shard [Test][ShardedTensor] Add test for corner case for chunk sharding spec (#109626) 2023-09-20 14:40:07 +00:00
_spmd Move has_triton to top level triton utils so that dynamo can also access (#109832) 2023-09-22 19:33:41 +00:00
_tensor [dtensor] support lt/gt op (#110585) 2023-10-07 00:06:36 +00:00
_tools add number of cuda retries into tracker (#92557) 2023-01-25 14:44:34 +00:00
algorithms [BE] Fix all B022 useless-contextlib-suppress (#100335) 2023-04-30 18:47:40 +00:00
bin
checkpoint [state_dict][1/N] Implement the basic functions of distributed.checkpoint._state_dict (#105902) 2023-10-05 20:04:15 +00:00
elastic [RESUBMIT] Standardize on error types for distributed errors. (#108191) 2023-08-30 21:47:39 +00:00
fsdp [FSDP] Change _create_chunk_dtensor in fsdp/_shard_utils.py to use public API from DTensor (#110831) 2023-10-10 21:04:27 +00:00
launcher removing some redundant str splits (#106089) 2023-09-01 00:22:58 +00:00
nn/jit Have test classes extend from common_utils.TestCase, not unittest.TestCase (#66900) 2021-10-19 16:54:05 -07:00
optim Back out "Reland "Make adding buffers more like adding parameters (#104069)" (#106224)" (#106743) 2023-08-08 15:27:34 +00:00
pipeline/sync [BE] Enable ruff's UP rules and autoformat distributed/ (#105433) 2023-07-19 14:27:11 +00:00
rpc Revert "Revert "Expandable blocks in allocator (#96995)"" (#99275) 2023-04-17 23:46:08 +00:00
tensor/parallel [TP][Inference] Enable DTensor TP inference (#110751) 2023-10-07 18:57:27 +00:00
argparse_util_test.py [skip ci] set more tests with owners for distributed and elastic (#67583) 2021-11-01 12:26:03 -07:00
test_c10d_common.py Add "cuda" to MPI backend capabilities (#109614) 2023-09-21 13:34:58 +00:00
test_c10d_gloo.py [RESUBMIT] Standardize on error types for distributed errors. (#108191) 2023-08-30 21:47:39 +00:00
test_c10d_logger.py [C10D][Logger]Add more info to c10d logger (#107331) 2023-08-28 15:10:56 +00:00
test_c10d_nccl.py [RESUBMIT] Standardize on error types for distributed errors. (#108191) 2023-08-30 21:47:39 +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_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_spawn.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
test_c10d_ucc.py [BE] Enable ruff's UP rules and autoformat distributed/ (#105433) 2023-07-19 14:27:11 +00:00
test_collective_utils.py Initial commit of collective_utils (#101037) 2023-06-27 02:15:16 +00:00
test_data_parallel.py Revert "[reland] Update custom Function preserve torch function when inputs r… (#110679)" 2023-10-09 19:09:01 +00:00
test_distributed_spawn.py Back out "Revert "[DDP] multiple forward support for static graph (#103487)" (#103873)" (#103938) 2023-06-22 21:55:58 +00:00
test_dynamo_distributed.py Move has_triton to top level triton utils so that dynamo can also access (#109832) 2023-09-22 19:33:41 +00:00
test_fake_pg.py Enhance fakepg: add fsdp+tp tests (#107626) 2023-08-25 06:17:54 +00:00
test_functional_api.py [collective] Add a torch.compile + functional_collectives test (#110688) 2023-10-10 17:14:50 +00:00
test_inductor_collectives.py Move export.constrain_as_* to torch._constrain_as_* (#110757) 2023-10-11 02:37:55 +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 [C10D] Improve MTPG autograd test. Fixes #105106 (#105356) 2023-07-20 13:51:21 +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 [c10d] Add tests for usig libuv through init_process_group. (#108661) 2023-09-20 16:02:20 +00:00