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