mltop 0.5.0

Resource monitor for ML engineers written in Rust
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
51
52
53

use sysinfo::{Components, System};

use std::collections::HashMap;

#[derive(Clone)]
pub struct CoreSnapshot {
    pub usage: f32,
    pub temp: f32,
}

#[derive(Clone)]
pub struct CpuSnapshot {
    pub usage: f32, // as a value between 0.0 and 100.0
    pub cores: Vec<CoreSnapshot>,
}

impl CpuSnapshot {
    pub fn from_sysinfo(sys: &System) -> CpuSnapshot {
        let mut cores: Vec<CoreSnapshot> = Vec::new();

        // TODO: Fix temperature mismatched for all cores. Have to find a more robust way
        // to find the 1:1 core temperatures.

        // gather temperatures from Components in to a map
        let components = Components::new_with_refreshed_list();
        let mut temperatures = HashMap::new();
        for component in &components {
            let label = component.label();
            if label.contains("Core") {
                if let Ok(id) = label.split_whitespace().last().unwrap().parse::<usize>() {
                    temperatures.insert(id, component.temperature());
                }
            }
        }

        for cpu in sys.cpus() {
            let id: usize = cpu.name()[3..].parse().unwrap();
            let usage: f32 = cpu.cpu_usage();
            let temperature: f32 = match temperatures.get(&id) {
                Some(t) => *t,
                None => 0.0,
            };
            let core = CoreSnapshot {
                usage,
                temp: temperature,
            };
            cores.push(core);
        }

        let usage = sys.global_cpu_info().cpu_usage();
        CpuSnapshot { usage, cores }
    }
}