Enum ComputeDevice 

pub enum ComputeDevice {
    Cpu(CpuInfo),
    Gpu(GpuInfo),
    Tpu(TpuInfo),
    AppleSilicon(AppleSiliconInfo),
}

Compute device abstraction

Variants

Cpu(CpuInfo)

CPU device

Gpu(GpuInfo)

GPU device (NVIDIA, AMD, Intel)

Tpu(TpuInfo)

TPU device (Google)

AppleSilicon(AppleSiliconInfo)

Apple Silicon with unified memory

Implementations

impl ComputeDevice

pub fn detect() -> Vec<Self>

Auto-detect available compute devices

pub fn is_gpu(&self) -> bool

Check if this is a GPU device

pub fn is_cpu(&self) -> bool

Check if this is a CPU device

pub fn is_tpu(&self) -> bool

Check if this is a TPU device

pub fn is_apple_silicon(&self) -> bool

Check if this is Apple Silicon

pub fn memory_bytes(&self) -> u64

Get available memory in bytes

pub fn name(&self) -> &str

Get device name

pub fn compute_units(&self) -> u32

Get compute cores/units

pub fn relative_compute_power(&self) -> f64

Estimate relative compute power (normalized, CPU = 1.0)

Trait Implementations

impl Clone for ComputeDevice

fn clone(&self) -> ComputeDevice

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ComputeDevice

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for ComputeDevice

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for ComputeDevice

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

Formats the value using the given formatter.

impl PartialEq for ComputeDevice

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

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for ComputeDevice