Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions src/array_api_stubs/_draft/array_object.py
Original file line number Diff line number Diff line change
Expand Up @@ -514,8 +514,15 @@ 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
"""

def __eq__(self: array, other: Union[int, float, complex, bool, array], /) -> array:
Expand Down
Loading