gpu-histop 0.1.0

High-resolution GPU history monitor for NVIDIA, AMDGPU, and Apple Silicon
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

use std::collections::VecDeque;
use std::time::{Duration, Instant};

use crate::model::GpuSample;

#[derive(Debug, Clone)]
pub struct History {
    retention: Duration,
    samples: VecDeque<GpuSample>,
}

impl History {
    pub fn new(retention: Duration) -> Self {
        Self {
            retention,
            samples: VecDeque::new(),
        }
    }

    pub fn push(&mut self, sample: GpuSample) {
        let newest = sample.at;
        self.samples.push_back(sample);
        self.prune(newest);
    }

    pub fn latest(&self) -> Option<&GpuSample> {
        self.samples.back()
    }

    pub fn len(&self) -> usize {
        self.samples.len()
    }

    pub fn is_empty(&self) -> bool {
        self.samples.is_empty()
    }

    pub fn iter_window(
        &self,
        now: Instant,
        window: Duration,
    ) -> impl DoubleEndedIterator<Item = &GpuSample> {
        self.samples
            .iter()
            .filter(move |sample| sample.at <= now && now.duration_since(sample.at) <= window)
    }

    pub fn retention(&self) -> Duration {
        self.retention
    }

    fn prune(&mut self, now: Instant) {
        while let Some(front) = self.samples.front() {
            if front.at <= now && now.duration_since(front.at) > self.retention {
                self.samples.pop_front();
            } else {
                break;
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::model::GpuSample;

    #[test]
    fn prunes_samples_older_than_retention() {
        let t0 = Instant::now();
        let mut history = History::new(Duration::from_secs(2));
        history.push(sample_at(t0));
        history.push(sample_at(t0 + Duration::from_secs(1)));
        history.push(sample_at(t0 + Duration::from_secs(3)));
        assert_eq!(history.len(), 2);
    }

    fn sample_at(at: Instant) -> GpuSample {
        GpuSample {
            gpu_id: 0,
            at,
            gpu_util_percent: Some(0.0),
            mem_util_percent: None,
            vram_used_bytes: None,
            vram_total_bytes: None,
            power_watts: None,
            power_limit_watts: None,
            temperature_celsius: None,
            fan_percent: None,
            graphics_clock_mhz: None,
            memory_clock_mhz: None,
            compute_processes: None,
            processes: Vec::new(),
        }
    }
}