1 files changed, 10 insertions, 24 deletions
diff --git a/src/main.rs b/src/main.rs
index d5a3dcd..00c0fcc 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,5 +1,5 @@
 mod bpf_skel;
-use benchmark::{BenchmarkScheduler, Mode};
+use benchmark::BenchmarkScheduler;
 pub use bpf_skel::*;
 pub mod bpf_intf;
 
@@ -12,12 +12,14 @@ mod scheduler;
 mod socket;
 
 #[rustfmt::skip]
+#[allow(clippy::unwrap_or_default)]
+#[allow(clippy::useless_asref)]
+#[allow(clippy::single_match)]
+#[allow(clippy::needless_lifetimes)]
 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;
 
@@ -37,36 +39,20 @@ fn main() -> Result<()> {
             Arg::new("benchmark")
                 .short('b')
                 .long("benchmark")
-                .help("Enable benchmarking mode. Use \"e\" for e cores and \"p\" for p cores")
+                .help("Enable benchmarking mode.")
                 .required(false)
-                .value_name("mode"),
+                .action(ArgAction::SetTrue),
         )
         .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");
+    let benchmark = matches.get_flag("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))?;
+    if benchmark {
+        let mut sched = BenchmarkScheduler::init(&mut open_object, log_path)?;
         sched.run()?;
         return Ok(());
     }