From 9ceb02c0f886a10fe8a620ad879512fa6639f6f6 Mon Sep 17 00:00:00 2001
From: Lennard Kittner <lennard@kittner.dev>
Date: Sun, 9 Mar 2025 15:36:26 +0100
Subject: Try to keep tasks on the same e-core

---
 src/e_core_selector.rs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'src/e_core_selector.rs')

diff --git a/src/e_core_selector.rs b/src/e_core_selector.rs
index 6952e80..251cbe8 100644
--- a/src/e_core_selector.rs
+++ b/src/e_core_selector.rs
@@ -1,10 +1,9 @@
 use std::ops::Range;
 
 pub trait ECoreSelector {
-    fn next_core(&mut self) -> i32;
+    fn next_core(&mut self, previous_cpu: i32) -> i32;
 }
 
-//TODO: better core selection e.g. try to keep task on the same core
 pub struct RoundRobinSelector {
     offset: u32,
     num_cores: u32,
@@ -22,8 +21,10 @@ impl RoundRobinSelector {
 }
 
 impl ECoreSelector for RoundRobinSelector {
-    // TODO: use task cpu if suitable
-    fn next_core(&mut self) -> i32 {
+    fn next_core(&mut self, previous_cpu: i32) -> i32 {
+        if (self.offset..(self.offset + self.num_cores)).contains(&(previous_cpu as u32)) {
+            return previous_cpu;
+        }
         self.last_used += 1;
         (self.offset + (self.last_used % self.num_cores)) as i32
     }
-- 
cgit v1.3.1