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
|
mod bpf_skel;
use benchmark::{BenchmarkScheduler, Mode};
pub use bpf_skel::*;
pub mod bpf_intf;
mod benchmark;
mod core_selector;
mod energy;
mod freq;
mod model;
mod scheduler;
mod socket;
#[rustfmt::skip]
mod bpf;
use anyhow::Result;
use burn::tensor::Tensor;
use clap::{Arg, ArgAction, Command};
use model::load_model;
use scheduler::Scheduler;
use std::mem::MaybeUninit;
type Pid = i32;
fn main() -> Result<()> {
let matches = Command::new("Energy User Space Scheduler")
.arg(
Arg::new("perf")
.short('p')
.long("perf")
.help("Use this flag to switch between the kernel module and perf")
.action(ArgAction::SetTrue)
.required(false),
)
.arg(
Arg::new("benchmark")
.short('b')
.long("benchmark")
.help("Enable benchmarking mode. Use \"e\" for e cores and \"p\" for p cores")
.required(false)
.value_name("mode"),
)
.get_matches();
let device = Default::default();
let model = load_model("perf.pt");
let tensor = Tensor::from_floats(
[
800., 90678., 54734., 153646., 20354478., 40948418.,
89103105.,
//5200., 148947., 322426., 498965., 62340773., 144451046., 41976480.,
],
&device,
);
let result = model.forward(tensor);
let energy: f32 = result.into_scalar();
println!("energy: {energy}");
// panic!();
let use_perf = matches.get_flag("perf");
let benchmark = matches.get_one::<String>("benchmark");
// Initialize and load the scheduler.
let mut open_object = MaybeUninit::uninit();
let log_path = "/tmp/logs.csv";
if let Some(mode) = benchmark {
let mode = mode.trim().chars().next().unwrap();
let mut sched = BenchmarkScheduler::init(&mut open_object, log_path, Mode::from(mode))?;
sched.run()?;
return Ok(());
}
loop {
let mut sched = Scheduler::init(&mut open_object, use_perf)?;
if !sched.run()?.should_restart() {
break;
}
}
Ok(())
}
|
