Struct DeviceInfo 

pub struct DeviceInfo {
    pub cpu_cores: u32,
    pub total_memory: u64,
    pub os_type: String,
    pub os_release: String,
    pub has_cuda: bool,
    pub has_metal: bool,
    pub has_rocm: bool,
    pub has_directml: bool,
    pub devices: Vec<Device>,
}

Fields

cpu_cores: u32

total_memory: u64

os_type: String

os_release: String

has_cuda: bool

has_metal: bool

has_rocm: bool

has_directml: bool

devices: Vec<Device>

Implementations

impl DeviceInfo

pub fn probe() -> Self

Probe device information (cached).

pub fn try_probe() -> Result<Self, DeviceProbeError>

Probe device information (not cached).

pub fn try_probe_with_timeout( timeout: Duration, ) -> Result<Self, DeviceProbeError>

Probe device information with a timeout applied to external commands.

pub fn best_gpu(&self) -> Option<&Device>

Select the “best” GPU using a simple heuristic.

Primary key: memory_mb. Tie-breaker: compute_capability when present.

pub fn best_gpu_with_preference( &self, preference: &GpuPreference, ) -> Option<&Device>

pub fn cuda_devices(&self) -> Vec<&Device>

pub fn get_best_provider(&self) -> String

Get the best available execution provider.

pub fn has_provider(&self, provider: &str) -> bool

Check if a specific provider is available.

Trait Implementations

impl Clone for DeviceInfo

fn clone(&self) -> DeviceInfo

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DeviceInfo

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for DeviceInfo

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for DeviceInfo

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

where __S: Serializer,

Serialize this value into the given Serde serializer.