dlpack_device — Python array API standard 2024.12 documentation (original) (raw)

• Array API standard
• Consortium for Python Data API Standards
• API specification
• Array object

array.__dlpack_device__() → Tuple[Enum, int]¶

Returns device type and device ID in DLPack format. Meant for use within from_dlpack().

Parameters:

self (array) – array instance.

Returns:

device (Tuple[Enum, int]) – a tuple (device_type, device_id) in DLPack format. Valid device type enum members are:

CPU = 1 CUDA = 2 CPU_PINNED = 3 OPENCL = 4 VULKAN = 7 METAL = 8 VPI = 9 ROCM = 10 CUDA_MANAGED = 13 ONE_API = 14