Struct DLDevice 

#[repr(C)]
pub struct DLDevice {
    pub device_type: u32,
    pub device_id: i32,
}

A device descriptor specifying where tensor data resides.

This corresponds to DLDevice in the DLPack specification.

Fields

device_type: u32

The device type (CPU, CUDA, etc.)

device_id: i32

The device ID (e.g., which GPU for multi-GPU systems)

Implementations

impl DLDevice

pub fn new(device_type: DLDeviceType, device_id: i32) -> Self

Create a new device descriptor.

pub fn device_type_enum(&self) -> Option<DLDeviceType>

Get the device type as an enum.

Returns None for unknown device types.

pub fn is_cuda(&self) -> bool

Check if this is a CUDA device.

pub fn is_cpu(&self) -> bool

Check if this is a CPU device.

pub fn is_cuda_host(&self) -> bool

Check if this is CUDA host (pinned) memory.

pub fn is_rocm(&self) -> bool

Check if this is a ROCm device.

pub fn is_metal(&self) -> bool

Check if this is a Metal device (Apple GPU).

Trait Implementations

impl Clone for DLDevice

fn clone(&self) -> DLDevice

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DLDevice

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

Formats the value using the given formatter.

impl Hash for DLDevice

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 DLDevice

fn eq(&self, other: &DLDevice) -> 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 DLDevice

impl Eq for DLDevice

impl StructuralPartialEq for DLDevice