Struct HardwareDevice 

pub struct HardwareDevice {
    pub backend: HardwareBackend,
    pub device_id: usize,
    pub name: String,
    pub total_memory: usize,
    pub available_memory: usize,
    pub compute_capability: Option<(u32, u32)>,
}

Hardware device information

Fields

backend: HardwareBackend

Backend type

device_id: usize

Device ID

name: String

Device name

total_memory: usize

Total memory in bytes

available_memory: usize

Available memory in bytes

compute_capability: Option<(u32, u32)>

Compute capability (for CUDA/ROCm)

Implementations

impl HardwareDevice

pub fn cpu() -> Self

Create a CPU device

pub fn cuda(device_id: usize) -> Result<Self>

Create a CUDA device

pub fn rocm(device_id: usize) -> Result<Self>

Create a ROCm device

pub fn metal(device_id: usize) -> Result<Self>

Create a Metal device

pub fn tpu(device_id: usize) -> Result<Self>

Create a TPU device

Trait Implementations

impl Clone for HardwareDevice

fn clone(&self) -> HardwareDevice

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for HardwareDevice

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

Formats the value using the given formatter.