pytorch

10f3abc6 - [DeviceIndex][3/N] Use DeviceIndex in more places (#119635)

[DeviceIndex][3/N] Use DeviceIndex in more places (#119635) Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/119635 Approved by: https://github.com/ezyang