From 76d8a896bbff612e3d2db27554891c71d28988af Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?D=C3=A1niel=20Buga?= <bugadani@gmail.com>
Date: Tue, 17 Dec 2024 18:51:22 +0100
Subject: [PATCH] Make poll_to_despawn a no_op

---
 embassy-executor/src/raw/mod.rs | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/embassy-executor/src/raw/mod.rs b/embassy-executor/src/raw/mod.rs
index 4a4ecf603..e38a2af66 100644
--- a/embassy-executor/src/raw/mod.rs
+++ b/embassy-executor/src/raw/mod.rs
@@ -161,10 +161,8 @@ pub struct TaskStorage<F: Future + 'static> {
     future: UninitCell<F>, // Valid if STATE_SPAWNED
 }
 
-unsafe fn poll_to_despawn(p: TaskRef) {
-    // The task's future has already been dropped, we just mark it as `!SPAWNED`.
-    let this = p.header();
-    this.state.despawn();
+unsafe fn poll_exited(_p: TaskRef) {
+    // Nothing to do, the task is already !SPAWNED and dequeued.
 }
 
 impl<F: Future + 'static> TaskStorage<F> {
@@ -221,7 +219,7 @@ impl<F: Future + 'static> TaskStorage<F> {
 
                 // We replace the poll_fn with a despawn function, so that the task is cleaned up
                 // when the executor polls it next.
-                this.raw.poll_fn.set(Some(poll_to_despawn));
+                this.raw.poll_fn.set(Some(poll_exited));
 
                 // Make sure we despawn last, so that other threads can only spawn the task
                 // after we're done with it.