pytorch/torch/testing/_internal/distributed
Wanchao Liang cf3a5160f8 [BE] move init_multigpu_helper to common_distributed (#67050)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67050

This PR moves init_multi_gpu_helper to common_distributed so that it could be shared by different distributed tests.
ghstack-source-id: 141370119

Test Plan: wait for ci.

Reviewed By: mrshenli

Differential Revision: D31842644

fbshipit-source-id: c7bad25d6cef9bdce7ad1fb6c60c1cad4b765702
2021-10-22 17:16:11 -07:00
..
_sharded_tensor [PyTorch] Make rearragement in sharded linear work as expected. (#66603) 2021-10-19 23:16:38 -07:00
nn Remove outdated warning about RecursiveScriptModule not being copiable (#64085) 2021-08-31 21:31:32 -07:00
pipeline [Pipe] Refactor convert_to_balance under non-test package. (#50860) 2021-01-28 12:10:21 -08:00
rpc move get_cycles_per_ms() to common_utils (#66798) 2021-10-18 14:04:09 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
ddp_under_dist_autograd_test.py Split out CUDA RPC tests (#55695) 2021-04-12 07:48:08 -07:00
distributed_test.py [BE] move init_multigpu_helper to common_distributed (#67050) 2021-10-22 17:16:11 -07:00
pipe_with_ddp_test.py Correct backend in pipe_with_ddp_test (#60123) 2021-06-17 13:43:01 -07:00
rpc_utils.py Run sparse tests only for TensorPipe agent. (#66600) 2021-10-14 11:08:15 -07:00