pytorch/test/distributed
2024-12-02 14:08:04 +00:00
..
_composable [ca] expose option to collect sizes as dynamic (#141153) 2024-11-22 19:26:27 +00:00
_shard Flip default on weights_only (#137602) 2024-11-04 18:30:29 +00:00
_tensor [dtensor][random] allow user to manual_seed different seed on device mesh; only sync RNG state in WORLD when manual_seed has not been called (#141223) 2024-11-29 07:59:34 +00:00
_tools Revert "ILP for auto FSDP wrapping (#140298)" 2024-12-02 14:08:04 +00:00
algorithms
bin
checkpoint
elastic
flight_recorder [FR] Polish the log message for dtype mismatch and don't exit when too many mismatch (#140451) 2024-11-13 07:24:53 +00:00
fsdp [DTensor][unpickler] Add DTensor related classes to allowed globals so we can still torch.load(DTensor) with weights_only=True (#139949) 2024-11-08 05:06:11 +00:00
launcher
nn/jit
optim
pipelining [pipelining] allow multiple backward grads (#140981) 2024-11-23 00:35:08 +00:00
rpc
tensor/parallel
argparse_util_test.py
test_backends.py API to retrieve default distributed backend from device (#140536) 2024-11-22 11:01:53 +00:00
test_c10d_common.py PyProcessGroup: support rank, world size, group name/desc overrides (#141529) 2024-11-26 20:56:57 +00:00
test_c10d_functional_native.py
test_c10d_gloo.py
test_c10d_logger.py [c10d] Switch all timer logging in c10d to wait_counter (#141154) 2024-11-21 01:10:11 +00:00
test_c10d_nccl.py [c10d] Test needs abort; otherwise will hang (#141509) 2024-11-27 05:47:17 +00:00
test_c10d_object_collectives.py
test_c10d_ops_nccl.py
test_c10d_pypg.py PyProcessGroup: support rank, world size, group name/desc overrides (#141529) 2024-11-26 20:56:57 +00:00
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
test_control_collectives.py
test_data_parallel.py Replace clone.detach with detach.clone (#140264) 2024-11-13 07:01:02 +00:00
test_device_mesh.py
test_distributed_spawn.py
test_dynamo_distributed.py [Dynamo][Distributed] Fix ProcessGroup getattr (#141638) 2024-11-27 21:42:33 +00:00
test_fake_pg.py
test_functional_api.py Generalization of distributed test cases for non-CUDA devices (#138216) 2024-11-18 09:38:00 +00:00
test_inductor_collectives.py Move Sympy printers to torch/utils/_sympy/printers.py (#140597) 2024-11-26 18:11:00 +00:00
test_launcher.py
test_multi_threaded_pg.py
test_nccl.py
test_pg_wrapper.py
test_store.py
test_symmetric_memory.py [torch/distributed] Make _SymmetricMemory.has_multicast_support() ret… (#141598) 2024-11-26 23:36:32 +00:00