pytorch
1b34238d - fix get device index if has _utils._get_device_index in privateuse1 (#108123)

Commit
1 year ago
fix get device index if has _utils._get_device_index in privateuse1 (#108123) **Get device index by torch.privateuse1._utils._get_device_index, if the metched exists.** Reason: Can only get device_index 0 if ```location``` such as 'privateuse1' before modify. Can get accurate deivce index use _get_device_index in this scenario. Pull Request resolved: https://github.com/pytorch/pytorch/pull/108123 Approved by: https://github.com/albanD
Author
Committer
Parents
Loading