Enum DLDeviceType 

#[repr(i32)]
pub enum DLDeviceType {
    Cpu = 1,
    Cuda = 2,
    CudaHost = 3,
    OpenCL = 4,
    Vulkan = 7,
    Metal = 8,
    Rocm = 10,
}

Device type codes used by DLPack.

These integer codes identify which physical device (CPU, CUDA, Metal, etc.) holds the tensor data.

Variants

Cpu = 1

Host CPU (device type 1).

Cuda = 2

CUDA GPU (device type 2).

CudaHost = 3

CUDA pinned host memory (device type 3).

OpenCL = 4

OpenCL device (device type 4).

Vulkan = 7

Vulkan device (device type 7).

Metal = 8

Apple Metal device (device type 8).

Rocm = 10

AMD ROCm/HIP GPU (device type 10).

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 Debug for DLDeviceType

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

Formats the value using the given formatter.

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 Copy for DLDeviceType

impl Eq for DLDeviceType

impl StructuralPartialEq for DLDeviceType