Enum ComputeDevice 

pub enum ComputeDevice {
    Cpu {
        cores: u32,
        threads_per_core: u32,
        architecture: CpuArchitecture,
    },
    Gpu {
        name: String,
        memory_gb: f32,
        compute_capability: Option<String>,
        vendor: GpuVendor,
    },
    Tpu {
        version: TpuVersion,
        cores: u32,
    },
    AppleSilicon {
        chip: AppleChip,
        neural_engine_cores: u32,
        gpu_cores: u32,
        memory_gb: u32,
    },
    Edge {
        name: String,
        power_budget_watts: f32,
    },
}

Compute device abstraction for heterogeneous hardware

Variants

Cpu

Standard CPU execution

Fields

cores: u32

threads_per_core: u32

architecture: CpuArchitecture

Gpu

NVIDIA/AMD GPU acceleration

Fields

name: String

memory_gb: f32

compute_capability: Option<String>

vendor: GpuVendor

Tpu

Google TPU accelerator

Fields

version: TpuVersion

cores: u32

AppleSilicon

Apple Silicon unified memory

Fields

chip: AppleChip

neural_engine_cores: u32

gpu_cores: u32

memory_gb: u32

Edge

Edge/embedded devices

Fields

name: String

power_budget_watts: f32

Implementations

impl ComputeDevice

pub fn theoretical_flops(&self) -> f64

Calculate theoretical FLOPS for the device

pub fn estimated_power_watts(&self) -> f32

Estimate power consumption in watts

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 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