diff --git a/src/array_api_stubs/_draft/array_object.py b/src/array_api_stubs/_draft/array_object.py index 9a2d36696..c4148025d 100644 --- a/src/array_api_stubs/_draft/array_object.py +++ b/src/array_api_stubs/_draft/array_object.py @@ -514,9 +514,17 @@ def __dlpack_device__(self: array, /) -> Tuple[Enum, int]: METAL = 8 VPI = 9 ROCM = 10 + ROCM_PINNED = 11 CUDA_MANAGED = 13 ONE_API = 14 - """ + WEB_GPU = 15 + HEXAGON = 16 + MAIA = 17 + TRAINIUM = 18 + TPU = 19 + TPU_PINNED = 20 + """ + # The ExtDev (enum value 12) device has not been included as its semantics are not fixed. def __eq__(self: array, other: Union[int, float, complex, bool, array], /) -> array: r"""