pytorch/test/distributed
2022-12-11 06:39:38 +00:00
..
_composable [FSDP] Allow MixedPrecision to skip inputs (#90620) 2022-12-11 06:39:38 +00:00
_shard Revert "remove torch.equal usages (#89527)" 2022-12-02 21:36:13 +00:00
_tensor [dtensor] remove torchgen function schema and parse manually (#90106) 2022-12-06 05:45:00 +00:00
_tools [reland] add save and load stats in memory_tracker (#90510) 2022-12-10 01:39:22 +00:00
algorithms prepare removal of deprecated functionality in torch.testing (#87969) 2022-11-02 14:04:48 +00:00
bin
checkpoint [Checkpoint][2D][6/N] Add optimizer and update default_planner to core distributed (#90212) 2022-12-08 02:53:29 +00:00
elastic Add watchdog to TorchElastic agent and trainers (#84081) 2022-09-07 00:17:20 +00:00
fsdp [FSDP] Allow MixedPrecision to skip inputs (#90620) 2022-12-11 06:39:38 +00:00
launcher
nn/jit
optim [PT-D][Easy] Reformat the optim code within PTD code base (#90399) 2022-12-08 06:38:59 +00:00
pipeline/sync Revert "remove torch.equal usages (#89527)" 2022-12-02 21:36:13 +00:00
rpc [ci] remove IN_CI env var 2022-06-11 17:16:30 +00:00
tensor/parallel [tp] ufmt test/distributed/tensor (#89970) 2022-12-01 20:58:16 +00:00
argparse_util_test.py
test_c10d_common.py [21/N] Add alltoall_base custom op with CPU/CUDA implementations (#89813) 2022-12-08 23:39:26 +00:00
test_c10d_error_logger.py [C10D][BE] Add exception handlers to c10d collectives function (#87643) (#87988) 2022-10-29 04:38:34 +00:00
test_c10d_gloo.py [21/N] Add alltoall_base custom op with CPU/CUDA implementations (#89813) 2022-12-08 23:39:26 +00:00
test_c10d_nccl.py [21/N] Add alltoall_base custom op with CPU/CUDA implementations (#89813) 2022-12-08 23:39:26 +00:00
test_c10d_object_collectives.py Revert "Revert "[distributed] Handle object collectives and NCCL. (#79034)"" 2022-06-15 10:04:37 -07:00
test_c10d_pypg.py [distributed] Make DDP work with python process group (#79176) 2022-06-28 17:14:21 +00:00
test_c10d_spawn_gloo.py
test_c10d_spawn_nccl.py Ensure tensors are contiguous in functional all_gather. 2022-06-17 01:27:11 +00:00
test_c10d_spawn_ucc.py add test_c10d_spawn_ucc.py (#86508) 2022-11-16 22:50:11 +00:00
test_c10d_spawn.py
test_data_parallel.py
test_distributed_spawn.py ProcessGroupUCC tests (#83285) 2022-09-10 10:56:05 +00:00
test_dynamo_distributed.py Fix AssertionError fake_mode is not None in distributed (#90392) 2022-12-07 20:12:39 +00:00
test_launcher.py
test_multi_threaded_pg.py [threaded PG] fix long hang issue in testing (#90515) 2022-12-09 05:24:08 +00:00
test_nccl.py
test_pg_wrapper.py Add _reduce_scatter_base to ProcessGroupWrapper. (#79633) 2022-06-29 15:32:42 +00:00
test_store.py Add reference counter in FileStore (#85601) 2022-10-07 17:59:29 +00:00