From 4fedf19e5175072c910845bfbb66fb34ecde1673 Mon Sep 17 00:00:00 2001
From: Ashwin Naren <arihant2math@gmail.com>
Date: Sun, 28 Dec 2025 00:58:52 -0800
Subject: [PATCH] Optimize current-cpu case in task insertion

---
 src/sched/mod.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/sched/mod.rs b/src/sched/mod.rs
index 8fd267f..07b2cd6 100644
--- a/src/sched/mod.rs
+++ b/src/sched/mod.rs
@@ -151,7 +151,11 @@ pub fn insert_task(task: Arc<Task>) {
 
 pub fn insert_task_cross_cpu(task: Arc<Task>) {
     let cpu = get_next_cpu();
-    message_cpu(cpu.value(), Message::PutTask(task)).expect("Failed to send task to CPU");
+    if cpu == CpuId::this() {
+        insert_task(task);
+    } else {
+        message_cpu(cpu.value(), Message::PutTask(task)).expect("Failed to send task to CPU");
+    }
 }
 
 pub struct SchedState {