pytorch

9a2a6fcf - add get_device_index for custom device (#98804)

add get_device_index for custom device (#98804) Fixes #ISSUE_NUMBER as the title. Pull Request resolved: https://github.com/pytorch/pytorch/pull/98804 Approved by: https://github.com/ngimel