oxmera-core 0.4.0

Core types for oxmera: dtype, shape, strides, layout, device handles, and the typed error taxonomy.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

//! Device handles.

/// Where a tensor's storage lives and where its work runs.
///
/// This is a *handle*, not a backend: it names a place. The backend
/// registry resolves a handle to an implementation; this crate must never
/// know how.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[non_exhaustive]
pub enum Device {
    /// The multi-threaded CPU backend — always available.
    Cpu,
    /// An Apple-Silicon GPU via Metal, by device index.
    Metal {
        /// Zero-based device index.
        index: usize,
    },
    /// An NVIDIA GPU, by device index — served by `oxmera-cuda` when a
    /// driver and a device are present at load time.
    Cuda {
        /// Zero-based device index.
        index: usize,
    },
}

impl Device {
    /// A short stable name for the device kind (`"cpu"`, `"metal"`,
    /// `"cuda"`), used in error messages and `oxmera doctor` output.
    pub fn kind_name(self) -> &'static str {
        match self {
            Device::Cpu => "cpu",
            Device::Metal { .. } => "metal",
            Device::Cuda { .. } => "cuda",
        }
    }
}