pytorch/test/distributed
Edward Z. Yang 0d9c9716b2 Ensure compiler collective is called even when no graph is compiled (#132163)
It's very important to make sure we always run the compiler collective, because if we don't, we will fail to apply automatic dynamic at all.

Signed-off-by: Edward Z. Yang <ezyang@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/132163
Approved by: https://github.com/jansel
2024-08-02 12:18:34 +00:00
..
_composable Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
_shard Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
_tensor Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
_tools Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
algorithms Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
bin
checkpoint Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
elastic Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
fsdp Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
launcher Validate logs are created by default (#128522) 2024-06-17 20:07:13 +00:00
nn/jit [BE][Easy] replace import pathlib with from pathlib import Path (#129426) 2024-06-30 01:36:07 +00:00
optim Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
pipelining [PP] Forward only schedule (#132177) 2024-08-01 16:35:56 +00:00
rpc [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
tensor/parallel Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
argparse_util_test.py
test_c10d_common.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_c10d_functional_native.py [inductor]Add DtypeView to avoid memory leak and unnecessary kernel generations (#128883) 2024-07-23 17:31:39 +00:00
test_c10d_gloo.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_c10d_logger.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_c10d_nccl.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_c10d_object_collectives.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_c10d_ops_nccl.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_c10d_pypg.py
test_c10d_spawn.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_c10d_spawn_gloo.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_c10d_spawn_nccl.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_c10d_spawn_ucc.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_c10d_ucc.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_collective_utils.py
test_compute_comm_reordering.py [Traceable FSDP2][Inductor] Create grouped nodes for FSDP2 all-gather code block and reduce-scatter code block (after Buffer/Operation split) (#131510) 2024-07-27 08:39:58 +00:00
test_control_collectives.py [c10d] add a simple test to demonstrate the user usage of collectives (#127665) 2024-06-05 04:32:11 +00:00
test_data_parallel.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_device_mesh.py [DeviceMesh] Remove _parent_mesh as an attribute from DeviceMesh and remove it from DeviceMesh's hash (#131636) 2024-07-25 22:47:22 +00:00
test_distributed_spawn.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_dynamo_distributed.py Ensure compiler collective is called even when no graph is compiled (#132163) 2024-08-02 12:18:34 +00:00
test_fake_pg.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_functional_api.py Only make wait_tensor as a side_effect op (#132341) 2024-08-02 01:24:40 +00:00
test_inductor_collectives.py Modify ExternKernelAlloc with NoneLayout to not assign its result to anything (#129188) 2024-06-22 02:57:44 +00:00
test_launcher.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_multi_threaded_pg.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_nccl.py [BE][Easy][10/19] enforce style for empty lines in import segments in test/d*/ (#129761) 2024-07-17 16:57:39 +00:00
test_pg_wrapper.py
test_store.py Add None return type to init -- tests rest (#132376) 2024-08-01 15:44:51 +00:00
test_symmetric_memory.py [micro_pipeline_tp] implement the pass for fused_scaled_matmul_reduce_scatter (#131951) 2024-07-30 23:02:49 +00:00