Struct DeviceManager 

pub struct DeviceManager { /* private fields */ }

Operation-level device manager wrapping a pluggable DeviceSelector.

DeviceManager is the public entry point for the device-selection framework. Callers construct one with a selector of their choice (or use DeviceManager::with_heuristic for the built-in heuristic), then call DeviceManager::select once per scheduled operation.

Examples

use tensorlogic_scirs_backend::device_manager::{
    DeviceConfig, DeviceManager, OpDescriptor, OpKind,
};

let mgr = DeviceManager::with_heuristic(DeviceConfig::default());
let op  = OpDescriptor { kind: OpKind::MatMul };
let dev = mgr.select(&op, &[32, 32]);
assert!(dev.is_cpu()); // GPU not available by default

Implementations

impl DeviceManager

pub fn new(selector: impl DeviceSelector + 'static) -> Self

Create a DeviceManager backed by any DeviceSelector implementation.

pub fn with_heuristic(config: DeviceConfig) -> Self

Create a DeviceManager backed by the built-in HeuristicSelector configured with config.

pub fn select(&self, op: &OpDescriptor, shape: &[usize]) -> Device

Select the compute device for an operation described by op acting on a tensor with the given shape.