pytorch/test/distributed
Simon Fan db4e8a1d8a [ca] expose option to collect sizes as dynamic (#141153)
This is to address recompiles from eager nodes that saved dynamic activations

Pull Request resolved: https://github.com/pytorch/pytorch/pull/141153
Approved by: https://github.com/jansel
ghstack dependencies: #141152
2024-11-22 19:26:27 +00:00
..
_composable [ca] expose option to collect sizes as dynamic (#141153) 2024-11-22 19:26:27 +00:00
_shard
_tensor Fix condition for weights_only unpickler for DTensor (#140740) 2024-11-19 02:44:53 +00:00
_tools
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
launcher
nn/jit
optim
pipelining Replace clone.detach with detach.clone (#140264) 2024-11-13 07:01:02 +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 [C10D] Support group_dst/group_src in c10d send/recv (#140460) 2024-11-16 14:24:45 +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] Relax CUDA context test criteria (#141298) 2024-11-22 18:38:25 +00:00
test_c10d_object_collectives.py
test_c10d_ops_nccl.py
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
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
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 Revert "Move Sympy printers to torch/utils/_sympy/printers.py (#140597)" 2024-11-22 15:43:39 +00:00
test_launcher.py
test_multi_threaded_pg.py
test_nccl.py
test_pg_wrapper.py
test_store.py
test_symmetric_memory.py Implements user buffer registration using MemPool (#133603) 2024-11-21 01:40:11 +00:00