test-smi-rs 0.1.2

APIs for managing and monitoring FuriosaAI NPUs
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
37
38

use crate::generated::binding::*;
use crate::util::c_char_slice_to_string;

pub(crate) fn new_device_file(raw: FuriosaSmiDeviceFile) -> DeviceFile {
    DeviceFile { raw }
}

/// A struct for device file
pub struct DeviceFile {
    raw: FuriosaSmiDeviceFile,
}

impl DeviceFile {
    /// Get a list of core for device file
    pub fn cores(&self) -> Vec<u32> {
        (self.raw.core_start..=self.raw.core_end).collect()
    }

    /// Get a device file path
    pub fn path(&self) -> String {
        c_char_slice_to_string(&self.raw.path)
    }
}

pub(crate) fn default_device_file() -> FuriosaSmiDeviceFile {
    FuriosaSmiDeviceFile {
        core_start: 0,
        core_end: 0,
        path: [0; FURIOSA_SMI_MAX_PATH_SIZE as usize],
    }
}

pub(crate) fn default_device_files() -> FuriosaSmiDeviceFiles {
    FuriosaSmiDeviceFiles {
        count: 0,
        device_files: [default_device_file(); FURIOSA_SMI_MAX_DEVICE_HANDLE_SIZE as usize],
    }
}