diff options
Diffstat (limited to 'src/main.rs')
-rw-r--r-- | src/main.rs | 22 |
1 files changed, 7 insertions, 15 deletions
diff --git a/src/main.rs b/src/main.rs index 9f3528c..d5a3dcd 100644 --- a/src/main.rs +++ b/src/main.rs @@ -4,7 +4,7 @@ pub use bpf_skel::*; pub mod bpf_intf; mod benchmark; -mod e_core_selector; +mod core_selector; mod energy; mod freq; mod model; @@ -26,10 +26,10 @@ type Pid = i32; fn main() -> Result<()> { let matches = Command::new("Energy User Space Scheduler") .arg( - Arg::new("mock") - .short('m') - .long("mock") - .help("Use this flag if to activate kernel module mocking") + Arg::new("perf") + .short('p') + .long("perf") + .help("Use this flag to switch between the kernel module and perf") .action(ArgAction::SetTrue) .required(false), ) @@ -41,13 +41,6 @@ fn main() -> Result<()> { .required(false) .value_name("mode"), ) - .arg( - Arg::new("power_cap") - .long("energy_cap") - .help("Set a power cap for the processor") - .required(false) - .value_name("power in watts"), - ) .get_matches(); let device = Default::default(); @@ -65,8 +58,7 @@ fn main() -> Result<()> { println!("energy: {energy}"); // panic!(); - let power_cap = *matches.get_one::<u64>("power_cap").unwrap_or(&u64::MAX); - let use_mocking = matches.get_flag("mock"); + let use_perf = matches.get_flag("perf"); let benchmark = matches.get_one::<String>("benchmark"); // Initialize and load the scheduler. @@ -79,7 +71,7 @@ fn main() -> Result<()> { return Ok(()); } loop { - let mut sched = Scheduler::init(&mut open_object, use_mocking, power_cap)?; + let mut sched = Scheduler::init(&mut open_object, use_perf)?; if !sched.run()?.should_restart() { break; } |