mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-06 09:17:11 +08:00
[BE] use DeviceIndex instead of int64_t for related device interfaces (#103068)
This PR unifies the device interfaces in aten/*cpp and torch/csrc/*cpp to use **c10::DeviceIndex**. Pull Request resolved: https://github.com/pytorch/pytorch/pull/103068 Approved by: https://github.com/malfet
This commit is contained in:
@ -1803,6 +1803,7 @@ class BaseTy(Enum):
|
||||
bool = auto()
|
||||
Layout = auto()
|
||||
Device = auto()
|
||||
DeviceIndex = auto()
|
||||
Scalar = auto()
|
||||
MemoryFormat = auto()
|
||||
QScheme = auto()
|
||||
|
||||
Reference in New Issue
Block a user