mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
We should only pass the `device_id` when the backend is `nccl`. Otherwise, we would run into the following error: ``` RuntimeError: No backend for the parent process group or its backend does not support splitting ``` This also fixes test failure is not asserted when using `with_comms()` or `with_comms(eager_init=False)`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/139097 Approved by: https://github.com/XilunWu |
||
|---|---|---|
| .. | ||
| _shard | ||
| _tensor | ||
| nn | ||
| rpc | ||
| __init__.py | ||
| checkpoint_utils.py | ||
| common_state_dict.py | ||
| ddp_under_dist_autograd_test.py | ||
| distributed_test.py | ||
| distributed_utils.py | ||
| fake_pg.py | ||
| multi_threaded_pg.py | ||
| rpc_utils.py | ||