Revert ExplicitDevice changes

cyyever · cyyever · commit daac1189bbb2 · 2025-05-13T09:43:32.000+08:00
diff --git a/torch/accelerator/__init__.py b/torch/accelerator/__init__.py
@@ -6,7 +6,6 @@
 from typing_extensions import deprecated
 
 import torch
-from torch.types import ExplicitDevice
 
 from ._utils import _device_t, _get_device_index
 
@@ -121,7 +120,7 @@ def current_device_index() -> int:
 )(current_device_index)
 
 
-def set_device_index(device: ExplicitDevice, /) -> None:
+def set_device_index(device: _device_t, /) -> None:
     r"""Set the current device index to a given device.
 
     Args:
diff --git a/torch/types.py b/torch/types.py
@@ -70,9 +70,8 @@
 
 # Meta-type for "device-like" things.  Not to be confused with 'device' (a
 # literal device object).  This nomenclature is consistent with PythonArgParser.
-ExplicitDevice: TypeAlias = Union[_device, str, int]
 # None means use the default device (typically CPU)
-Device: TypeAlias = Union[ExplicitDevice, None]
+Device: TypeAlias = Union[_device, str, int, None]
 
 
 # Storage protocol implemented by ${Type}StorageBase classes
