| .. |
|
_composable
|
[FSDP2] Computed grad divide factors at runtime (#125484)
|
2024-05-05 14:11:33 +00:00 |
|
_shard
|
|
|
|
_spmd
|
|
|
|
_tensor
|
Revert "try to fix the warning in distribute_tensor (#125476)"
|
2024-05-04 22:25:32 +00:00 |
|
_tools
|
|
|
|
algorithms
|
|
|
|
bin
|
|
|
|
checkpoint
|
[DCP] Adds strict option to DefaultPlanner (#123869)
|
2024-05-02 22:50:32 +00:00 |
|
elastic
|
Fix AttributeError when doing mock patch for FileTimerServerTest.test_expired_timers (#125144)
|
2024-05-01 12:08:04 +00:00 |
|
fsdp
|
[comm_mode] adding some initial c10d ops to CommDebugMode (#125475)
|
2024-05-04 04:20:46 +00:00 |
|
launcher
|
torchelastic: change monitor_interval default to 0.1 (#124692)
|
2024-04-24 01:44:41 +00:00 |
|
nn/jit
|
|
|
|
optim
|
|
|
|
pipeline/sync
|
[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)
|
2024-04-17 19:29:34 +00:00 |
|
pipelining
|
[pipelining] Add tests for tracing frontend (#125449)
|
2024-05-06 20:44:56 +00:00 |
|
rpc
|
|
|
|
tensor/parallel
|
[comm_mode] adding some initial c10d ops to CommDebugMode (#125475)
|
2024-05-04 04:20:46 +00:00 |
|
argparse_util_test.py
|
|
|
|
test_c10d_common.py
|
|
|
|
test_c10d_functional_native.py
|
switch tests from constrain_as* to torch._check* (#125253)
|
2024-05-01 21:01:27 +00:00 |
|
test_c10d_gloo.py
|
|
|
|
test_c10d_logger.py
|
|
|
|
test_c10d_nccl.py
|
[Distributed] Add P2P versions of *object_list operations (#124379)
|
2024-05-03 23:22:58 +00:00 |
|
test_c10d_object_collectives.py
|
[Distributed] Add P2P versions of *object_list operations (#124379)
|
2024-05-03 23:22:58 +00:00 |
|
test_c10d_pypg.py
|
|
|
|
test_c10d_spawn.py
|
|
|
|
test_c10d_spawn_gloo.py
|
|
|
|
test_c10d_spawn_nccl.py
|
|
|
|
test_c10d_spawn_ucc.py
|
|
|
|
test_c10d_ucc.py
|
|
|
|
test_collective_utils.py
|
|
|
|
test_compute_comm_reordering.py
|
Remove Inductor IRs for legacy functional collectives (#124992)
|
2024-05-05 19:49:58 +00:00 |
|
test_data_parallel.py
|
|
|
|
test_device_mesh.py
|
[dtensor] delete the old unused mesh_alltoall (#124879)
|
2024-04-30 18:30:34 +00:00 |
|
test_distributed_spawn.py
|
|
|
|
test_dynamo_distributed.py
|
[guards][cpp-guards] Optimize NN module getattr guards (#124522)
|
2024-05-04 22:08:56 +00:00 |
|
test_fake_pg.py
|
|
|
|
test_functional_api.py
|
distributed: templated ring attention (#124215)
|
2024-04-19 00:57:08 +00:00 |
|
test_inductor_collectives.py
|
switch tests from constrain_as* to torch._check* (#125253)
|
2024-05-01 21:01:27 +00:00 |
|
test_launcher.py
|
|
|
|
test_multi_threaded_pg.py
|
|
|
|
test_nccl.py
|
|
|
|
test_pg_wrapper.py
|
[torch/distributed] Check gloo availability when doing isinstance(pg,… (#124233)
|
2024-04-19 04:07:00 +00:00 |
|
test_store.py
|
|
|
