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