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

Number of physical cores

threads_per_core: u32

Threads per core (hyperthreading)

architecture: CpuArchitecture

CPU architecture

Gpu

NVIDIA/AMD GPU acceleration

Fields

name: String

GPU model name

memory_gb: f32

GPU memory in GB

compute_capability: Option<String>

Compute capability (e.g., “8.6” for Ampere)

vendor: GpuVendor

GPU vendor

Tpu

Google TPU accelerator

Fields

version: TpuVersion

TPU version

cores: u32

Number of TPU cores

AppleSilicon

Apple Silicon unified memory

Fields

chip: AppleChip

Apple chip model

neural_engine_cores: u32

Neural engine cores

gpu_cores: u32

GPU cores

memory_gb: u32

Unified memory in GB

Edge

Edge/embedded devices

Fields

name: String

Device name

power_budget_watts: f32

Power budget in watts

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

pub fn default_cpu() -> Self

Create a default CPU device based on current system

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 Default for ComputeDevice

fn default() -> Self

Returns the “default value” for a type.

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