hmll 0.1.0-rc.1

Safe, idiomatic Rust bindings to the hmll library for high-performance ML model loading
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50

//! Device types for specifying where data should be loaded.

use hmll_sys::{hmll_device, HMLL_DEVICE_CPU, HMLL_DEVICE_CUDA};

/// Represents a device where data can be loaded.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum Device {
    /// CPU memory
    Cpu,
    /// CUDA GPU memory
    Cuda,
}

impl Device {
    /// Convert to the underlying C enum value.
    #[inline(always)]
    pub(crate) const fn to_raw(self) -> hmll_device {
        match self {
            Device::Cpu => HMLL_DEVICE_CPU,
            Device::Cuda => HMLL_DEVICE_CUDA,
        }
    }

    /// Convert from the underlying C enum value.
    #[allow(dead_code)]
    #[inline(always)]
    pub(crate) const fn from_raw(device: hmll_device) -> Self {
        match device {
            HMLL_DEVICE_CPU => Device::Cpu,
            HMLL_DEVICE_CUDA => Device::Cuda,
        }
    }
}

impl Default for Device {
    /// Default device is CPU.
    #[inline(always)]
    fn default() -> Self {
        Device::Cpu
    }
}

impl std::fmt::Display for Device {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Device::Cpu => write!(f, "CPU"),
            Device::Cuda => write!(f, "CUDA"),
        }
    }
}