| .. |
|
_composable
|
[fsdp2] Accomodate FSDP2 to accept parent mesh > 2 (#125778)
|
2024-05-09 05:02:21 +00:00 |
|
_shard
|
|
|
|
_spmd
|
|
|
|
_tensor
|
[RELAND] Switch default behavoir of export IR to be predispatch (#125860)
|
2024-05-10 17:36:53 +00:00 |
|
_tools
|
|
|
|
algorithms
|
|
|
|
bin
|
|
|
|
checkpoint
|
[DSD] Implement broadcast_from_rank0 option for optim state_dict (#125339)
|
2024-05-08 07:22:20 +00:00 |
|
elastic
|
Prevent rendezvous shutdown on worker restarts (#124819)
|
2024-05-09 02:40:31 +00:00 |
|
fsdp
|
[comm_mode] adding some initial c10d ops to CommDebugMode (#125475)
|
2024-05-04 04:20:46 +00:00 |
|
launcher
|
|
|
|
nn/jit
|
|
|
|
optim
|
|
|
|
pipeline/sync
|
|
|
|
pipelining
|
[pipelining] Add pipeline schedules (#125975)
|
2024-05-11 21:17:53 +00:00 |
|
rpc
|
|
|
|
tensor/parallel
|
[dtensor] run transformer sdpa in dtensor (#122997)
|
2024-05-08 17:08:47 +00:00 |
|
argparse_util_test.py
|
|
|
|
test_c10d_common.py
|
[BE]: Update ruff to v0.4.4 (#125031)
|
2024-05-12 20:02:37 +00:00 |
|
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
|
[BE]: Update ruff to v0.4.4 (#125031)
|
2024-05-12 20:02:37 +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_ops_nccl.py
|
[c10d] Reduce test time by reusing ProcessGroup (#125648)
|
2024-05-08 22:33:40 +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
|
[dynamo][fsdp] Use Tensor match for FSDP modules (#125827)
|
2024-05-09 21:26:15 +00:00 |
|
test_fake_pg.py
|
|
|
|
test_functional_api.py
|
|
|
|
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
|
|
|
|
test_store.py
|
|
|
