summaryrefslogtreecommitdiff
path: root/src/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs22
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;
}