Struct DeviceConfig 

pub struct DeviceConfig { /* private fields */ }

Configuration for the built-in HeuristicSelector.

Use the builder methods to customise thresholds and forced-device overrides.

Examples

use tensorlogic_scirs_backend::device_manager::DeviceConfig;

// Enable GPU when tensors have ≥ 4 M elements
let cfg = DeviceConfig::default()
    .with_gpu_available(true)
    .with_gpu_threshold(4_194_304);

Implementations

impl DeviceConfig

pub fn with_gpu_threshold(self, n: usize) -> Self

Set the element count threshold above which GPU execution is considered.

Tensors with fewer than n elements will always run on CPU regardless of GPU availability.

pub fn with_gpu_available(self, avail: bool) -> Self

Declare whether a GPU is available on the current system.

pub fn force_cpu(self) -> Self

Force all operations to run on CPU, overriding every other setting.

pub fn force_gpu(self, idx: u32) -> Self

Force all operations to run on the GPU with the given device index, overriding every other setting.

pub fn with_gpu_index(self, idx: u32) -> Self

Set the GPU device index used when GPU execution is selected.

Trait Implementations

impl Clone for DeviceConfig

fn clone(&self) -> DeviceConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DeviceConfig

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

Formats the value using the given formatter.

impl Default for DeviceConfig

fn default() -> Self

Returns the “default value” for a type.