tvm_ffi.DLDeviceType

class tvm_ffi.DLDeviceType(value, names=_not_given, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: IntEnum

Enumeration mirroring DLPack’s DLDeviceType.

Values can be compared against Device.dlpack_device_type().

Examples

dev = tvm_ffi.device("cuda", 0)
assert dev.dlpack_device_type() == tvm_ffi.DLDeviceType.kDLCUDA

__init__(*args, **kwds)

Defined in Enum as method __init__().

Methods

__init__(*args, **kwds)

Defined in Enum as method __init__().

Attributes

kDLCPU
kDLCUDA
kDLCUDAHost
kDLOpenCL
kDLVulkan
kDLMetal
kDLVPI
kDLROCM
kDLROCMHost
kDLExtDev
kDLCUDAManaged
kDLOneAPI
kDLWebGPU
kDLHexagon
kDLTrn

kDLCPU = 1

kDLCUDA = 2

kDLCUDAHost = 3

kDLOpenCL = 4

kDLVulkan = 7

kDLMetal = 8

kDLVPI = 9

kDLROCM = 10

kDLROCMHost = 11

kDLExtDev = 12

kDLCUDAManaged = 13

kDLOneAPI = 14

kDLWebGPU = 15

kDLHexagon = 16

kDLTrn = 17