Call sched_state_task_picked() from pick_next_task_stop()

Otherwise, the scheduler state machine becomes confused (and goes into
a rescheduling loop) when stop-machine is triggered.

1 files changed, 9 insertions, 0 deletions

diff --git a/kernel/sched/stop_task.c b/kernel/sched/stop_task.c
index 604297a08b3a..3191f86ec0f0 100644
--- a/kernel/sched/stop_task.c
+++ b/kernel/sched/stop_task.c
@@ -1,5 +1,7 @@
 #include "sched.h"
 
+#include <litmus/preempt.h>
+
 /*
  * stop-task scheduling class.
  *
@@ -35,6 +37,13 @@ pick_next_task_stop(struct rq *rq, struct task_struct *prev, struct pin_cookie c
 
         stop->se.exec_start = rq_clock_task(rq);
 
+        /* Let the LITMUS^RT scheduler state machine know
+         * that a task was picked. This is needed because the
+         * LITMUS^RT scheduling plugin will not be called
+         * if the stop-task class picks a task.
+         */
+        sched_state_task_picked();
+
         return stop;
 }