We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 346c59a commit dbe38b8Copy full SHA for dbe38b8
1 file changed
array_api_strict/_devices.py
@@ -43,12 +43,14 @@ def _supported_dtypes(self) -> list[DType]:
43
class DLDeviceType(IntEnum):
44
kDLCPU = 1
45
kDLCUDA = 2
46
+ kDLMETAL = 8
47
48
49
_DLPACK_DEVICE_FOR: Final[dict[Device, tuple[DLDeviceType, int]]] = {
50
CPU_DEVICE: (DLDeviceType.kDLCPU, 0),
51
Device("device1"): (DLDeviceType.kDLCUDA, 0),
52
Device("device2"): (DLDeviceType.kDLCUDA, 1),
53
+ _F32_DEVICE: (DLDeviceType.kDLMETAL, 0),
54
}
55
56
_DLPACK_DEVICE_TO_LOGICAL: Final[dict[tuple[int, int], Device]] = {
0 commit comments