energy-bench 0.2.5

Methods for benchmarking the energy consumption of programs.
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

use indexmap::indexmap;
use rapl_energy::Rapl;

pub trait EnergyAccumulator {
    /// Gets the difference in value since the last time this energy probe was created/reset.
    fn elapsed(&self) -> indexmap::IndexMap<String, f32>;

    /// Resets this probe, such that the next time `elapsed` is called,
    /// the difference compared to the value at this reset is returned.
    fn reset(&mut self);
}

pub struct DefaultEnergyAccumulator {
    rapl: Option<Rapl>,
}

impl DefaultEnergyAccumulator {
    pub fn new() -> Self {
        let rapl = Rapl::now(true);
        if rapl.as_ref().is_none_or(|r| r.packages.is_empty()) {
            log::warn!("No RAPL packages were found. Please ensure you have read access to the files in `/sys/class/powercap/intel-rapl`.");
        }
        Self { rapl }
    }
}

impl EnergyAccumulator for DefaultEnergyAccumulator {
    fn elapsed(&self) -> indexmap::IndexMap<String, f32> {
        self.rapl.as_ref().map_or(indexmap! {}, |x| x.elapsed())
    }

    fn reset(&mut self) {
        if let Some(rapl) = &mut self.rapl {
            rapl.reset();
        }
    }
}