pytorch/test/distributed
2024-10-11 23:42:57 +00:00
..
_composable [pipelining] Shape Inference (#136912) 2024-10-11 22:49:00 +00:00
_shard
_tensor Add AsyncCollectiveTensor isinstance check to test_graph_input_is_async (#137253) 2024-10-09 08:06:16 +00:00
_tools
algorithms Fix ROCm skip decorator for test_ddp_tp and multiprocess UTs (#136161) 2024-09-18 11:01:23 +00:00
bin
checkpoint [distributed][test] Remove unused variable and fix doc typo (#136943) 2024-10-02 08:31:53 +00:00
elastic Fix rendezvous error due to EtcdStore get method not waiting in some cases (#137056) 2024-10-02 01:45:00 +00:00
flight_recorder
fsdp Enable channels_last format for FSDP (#137382) 2024-10-11 03:47:16 +00:00
launcher
nn/jit
optim Fix ROCm skip decorator for test_ddp_tp and multiprocess UTs (#136161) 2024-09-18 11:01:23 +00:00
pipelining [pipelining] Shape Inference (#136912) 2024-10-11 22:49:00 +00:00
rpc
tensor/parallel Allow parallelize_module to get device_mesh from ambient context (#134247) 2024-10-09 00:19:03 +00:00
argparse_util_test.py
test_c10d_common.py Fix prefix store seg fault (#136872) 2024-09-30 20:43:08 +00:00
test_c10d_functional_native.py Fix ir._WaitKernel (#137401) 2024-10-09 04:02:30 +00:00
test_c10d_gloo.py
test_c10d_logger.py
test_c10d_nccl.py Revert "[Distributed] Fix extra context on device 0 (#135273)" 2024-10-10 23:47: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 Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
test_control_collectives.py
test_data_parallel.py Using device-agnostic autocast api (#136613) 2024-09-27 07:16:24 +00:00
test_device_mesh.py [DeviceMesh] Fixed from_group when passing tensor mesh (#137713) 2024-10-11 14:53:51 +00:00
test_distributed_spawn.py
test_dynamo_distributed.py [distributed][test] Remove unused variable and fix doc typo (#136943) 2024-10-02 08:31:53 +00:00
test_fake_pg.py
test_functional_api.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
test_inductor_collectives.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
test_launcher.py
test_multi_threaded_pg.py
test_nccl.py
test_pg_wrapper.py
test_store.py [TCPStore] Throw value error if passing world_size=0 to TCPStore (#137792) 2024-10-11 23:42:57 +00:00
test_symmetric_memory.py [SymmetricMemory] set the storage_offset of tensors returned by get_buffer() to 0 (#137569) 2024-10-10 05:05:58 +00:00