pytorch

cb0886ec - [DeviceIndex][4/N] Use DeviceIndex in more places (#119741)

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