GSN/PSN: Update processor state on task exit

Without this change, a BUG_ON() in schedule() triggers.

2 files changed, 6 insertions, 0 deletions

diff --git a/litmus/sched_gsn_edf.c b/litmus/sched_gsn_edf.c
index fb2200bc93..1d84591c9c 100644
--- a/litmus/sched_gsn_edf.c
+++ b/litmus/sched_gsn_edf.c
@@ -583,6 +583,10 @@ static void gsnedf_task_exit(struct task_struct * t)
         /* unlink if necessary */
         spin_lock_irqsave(&gsnedf_lock, flags);
         unlink(t);
+        if (tsk_rt(t)->scheduled_on != NO_CPU) {
+                gsnedf_cpus[tsk_rt(t)->scheduled_on]->scheduled = NULL;
+                tsk_rt(t)->scheduled_on = NO_CPU;
+        }
         spin_unlock_irqrestore(&gsnedf_lock, flags);
 
         BUG_ON(!is_realtime(t));
diff --git a/litmus/sched_psn_edf.c b/litmus/sched_psn_edf.c
index a36a350160..42b5665c44 100644
--- a/litmus/sched_psn_edf.c
+++ b/litmus/sched_psn_edf.c
@@ -288,6 +288,8 @@ static void psnedf_task_exit(struct task_struct * t)
                 edf  = task_edf(t);
                 remove(edf, t);
         }
+        if (pedf->scheduled == t)
+                pedf->scheduled = NULL;
         preempt(pedf);
         spin_unlock_irqrestore(&pedf->slock, flags);
 }