| .. |
|
_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 |