1 files changed, 7 insertions, 0 deletions
diff --git a/src/scheduler.rs b/src/scheduler.rs
index 39db5c6..a9cad61 100644
--- a/src/scheduler.rs
+++ b/src/scheduler.rs
@@ -44,6 +44,7 @@ impl<'a> Scheduler<'a> {
         open_object: &'a mut MaybeUninit<OpenObject>,
         use_mocking: bool,
         power_cap: u64,
+        pids_to_monitor: &[Pid],
     ) -> Result<Self> {
         let bpf = BpfScheduler::init(
             open_object,
@@ -74,6 +75,12 @@ impl<'a> Scheduler<'a> {
             shared_cpu_current_frequencies.clone(),
         )?;
 
+        for pid in pids_to_monitor {
+            energy_sender
+                .try_send(EnergyRequest::LogEnergyTask(*pid))
+                .unwrap();
+        }
+
         let topology = Topology::new().unwrap();
         let mut e_core_ids = Vec::new();
         let mut p_core_ids = Vec::new();