Enum DLDeviceType 

#[repr(u32)]
pub enum DLDeviceType {
    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,
    kDLMAIA = 17,
    kDLTrn = 18,
}

Variants

kDLCPU = 1

\brief CPU device

kDLCUDA = 2

\brief CUDA GPU device

kDLCUDAHost = 3

\brief Pinned CUDA CPU memory by cudaMallocHost

kDLOpenCL = 4

\brief OpenCL devices.

kDLVulkan = 7

\brief Vulkan buffer for next generation graphics.

kDLMetal = 8

\brief Metal for Apple GPU.

kDLVPI = 9

\brief Verilog simulator buffer

kDLROCM = 10

\brief ROCm GPUs for AMD GPUs

kDLROCMHost = 11

\brief Pinned ROCm CPU memory allocated by hipMallocHost

kDLExtDev = 12

\brief Reserved extension device type, used for quickly test extension device The semantics can differ depending on the implementation.

kDLCUDAManaged = 13

\brief CUDA managed/unified memory allocated by cudaMallocManaged

kDLOneAPI = 14

\brief Unified shared memory allocated on a oneAPI non-partititioned device. Call to oneAPI runtime is required to determine the device type, the USM allocation type and the sycl context it is bound to.

kDLWebGPU = 15

\brief GPU support for next generation WebGPU standard.

kDLHexagon = 16

\brief Qualcomm Hexagon DSP

kDLMAIA = 17

\brief Microsoft MAIA devices

kDLTrn = 18

\brief AWS Trainium

Trait Implementations

impl Clone for DLDeviceType

fn clone(&self) -> DLDeviceType

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for DLDeviceType

impl Debug for DLDeviceType

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Eq for DLDeviceType

impl Hash for DLDeviceType

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for DLDeviceType

fn eq(&self, other: &DLDeviceType) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for DLDeviceType