pytorch/test/distributed
Will Constable 9fb9219478 Make DDPOptimizer work with torch._dynamo.explain() (#94749)
GraphModules that were created during DDPOptimizer graph breaking
lacked `compile_subgraph_reason`, which caused an exception when
running .explain().

Now the reason is provided and users can use .explain() to find out
that DDPOptimizer is causing graph breaks.

Fixes #94579

Pull Request resolved: https://github.com/pytorch/pytorch/pull/94749
Approved by: https://github.com/voznesenskym
2023-02-14 01:33:47 +00:00
..
_composable [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
_shard [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
_tensor [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
_tools add number of cuda retries into tracker (#92557) 2023-01-25 14:44:34 +00:00
algorithms [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
bin
checkpoint [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
elastic [BE]: Apply pyupgrade yield from and unit test alias upgrades (#94309) 2023-02-07 20:08:58 +00:00
fsdp [FSDP][1/N] Refactor module materialization (#94196) 2023-02-13 21:43:00 +00:00
launcher [BE] Prefer dash over underscore in command-line options (#94505) 2023-02-09 20:16:49 +00:00
nn/jit
optim [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
pipeline/sync [BE] Apply almost all remaining flake8-comprehension checks (#94676) 2023-02-12 01:01:25 +00:00
rpc [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
tensor/parallel [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
argparse_util_test.py
test_c10d_common.py Update documentation init_process_group optional backend (#94543) 2023-02-13 21:45:38 +00:00
test_c10d_error_logger.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_c10d_gloo.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_c10d_nccl.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_c10d_object_collectives.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_c10d_pypg.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_c10d_spawn.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_c10d_spawn_gloo.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_c10d_spawn_nccl.py
test_c10d_spawn_ucc.py Allow Process Group to support multiple backends (#88330) (#90997) 2022-12-16 23:15:00 +00:00
test_data_parallel.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_distributed_spawn.py correct use of torch.backends.cudnn.flags() (#93182) 2023-01-28 06:50:06 +00:00
test_dynamo_distributed.py Make DDPOptimizer work with torch._dynamo.explain() (#94749) 2023-02-14 01:33:47 +00:00
test_launcher.py [BE] Prefer dash over underscore in command-line options (#94505) 2023-02-09 20:16:49 +00:00
test_multi_threaded_pg.py [threaded_pg] full rewrite of MultiThreadedTestCase to enable device_type tests (#91650) 2023-01-17 03:26:36 +00:00
test_nccl.py [CUDA] Drop CUDA < 11.0 test flags (#92605) 2023-01-24 04:34:06 +00:00
test_pg_wrapper.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_store.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00