pytorch
05d11281 - [c10d] Start deprecating *_multigpu APIs (#85961)

Commit
2 years ago
[c10d] Start deprecating *_multigpu APIs (#85961) ### Deprecation reasons: - For most users training is on one GPU per process so these APIs are rarely used - They added one more API dimension - They can be expressed in a composed manner - They are not abstracted – specific to GPU - They caused backend APIs and implementations to have nested `std::vector<std::vector<Tensor>>`, which is hard to read or maintain Pull Request resolved: https://github.com/pytorch/pytorch/pull/85961 Approved by: https://github.com/XilunWu, https://github.com/H-Huang
Author
Committer
Parents
Loading