Dmitry Rogozhkin 7314cf44ae torch/accelerator: fix device type comparison (#143541) ...

This was failing without the fix:
```
python -c 'import torch; d=torch.device("xpu:0"); torch.accelerator.current_stream(d)'
```
with:
```
ValueError: xpu doesn't match the current accelerator xpu.
```

CC: @guangyey, @EikanWang

Pull Request resolved: https://github.com/pytorch/pytorch/pull/143541
Approved by: https://github.com/guangyey, https://github.com/albanD

2024-12-23 10:54:53 +00:00

..

__init__.py

[BE][accelerator] formalize API name {current,set}_device_{idx => index} (#140542)

2024-12-12 10:53:48 +00:00

_utils.py

torch/accelerator: fix device type comparison (#143541)

2024-12-23 10:54:53 +00:00