mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-08 07:39:33 +01:00
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 |
||
|---|---|---|
| .. | ||
| _sharded_tensor | ||
| nn | ||
| pipeline | ||
| rpc | ||
| __init__.py | ||
| ddp_under_dist_autograd_test.py | ||
| distributed_test.py | ||
| pipe_with_ddp_test.py | ||
| rpc_utils.py | ||