tvm_ffi.Device

class tvm_ffi.Device

Bases: object

A device descriptor used by TVM FFI and DLPack.

A Device identifies a placement (e.g. CPU, CUDA GPU) and a device index within that placement. Most users construct devices using tvm_ffi.device().

Parameters:

• device_type (Union[str, int]) – A device type name (e.g. "cpu", "cuda") or a DLPack device type code.

• index (int, optional) – Zero-based device index (defaults to 0 when omitted).

Examples

dev = tvm_ffi.device("cuda:0")
assert dev.type == "cuda"
assert dev.index == 0
assert str(dev) == "cuda:0"

__init__(*args, **kwargs)

Methods

__device_type_name__()	Return the canonical device type name (e.g. "cuda").
__init__(*args, **kwargs)
dlpack_device_type()	Return the corresponding DLDeviceType enum value.

Attributes

index	Zero-based device index.
type	Device type name such as "cpu" or "cuda".

__device_type_name__()

Return the canonical device type name (e.g. "cuda").

Return type:

str

dlpack_device_type()

Return the corresponding DLDeviceType enum value.

Return type:

int

index

Zero-based device index.

type

Device type name such as "cpu" or "cuda".