1 files changed, 29 insertions, 2 deletions

diff --git a/kernel/sched.c b/kernel/sched.c
index e4cfd193e76b..3682adadf6a7 100644
--- a/kernel/sched.c
+++ b/kernel/sched.c
@@ -79,6 +79,7 @@
 #include "sched_cpupri.h"
 #include "workqueue_sched.h"
 
+#include <litmus/litmus.h>
 #include <litmus/debug_trace.h>
 #include <litmus/sched_trace.h>
 #include <litmus/trace.h>
@@ -462,6 +463,8 @@ struct rq {
         /* runqueue lock: */
         raw_spinlock_t lock;
 
+        int dbg_in_schedule;
+
         /*
          * nr_running and cpu_load should be in the same cacheline because
          * remote CPUs use both these fields when doing load calculation.
@@ -573,6 +576,8 @@ static DEFINE_PER_CPU_SHARED_ALIGNED(struct rq, runqueues);
 static inline
 void check_preempt_curr(struct rq *rq, struct task_struct *p, int flags)
 {
+        if (test_tsk_need_resched(p))
+                TRACE_TASK(p, "already need_resched when resuming\n");
         rq->curr->sched_class->check_preempt_curr(rq, p, flags);
 
         /*
@@ -1193,6 +1198,10 @@ static void resched_task(struct task_struct *p)
         if (test_tsk_need_resched(p))
                 return;
 
+        if(task_rq(p)->dbg_in_schedule)
+                TRACE_TASK(p, "XXXX calling resched_task() during schedule() from ret:0x%p [rt:%d nr_running:%d]\n",
+                           __builtin_return_address(0), is_realtime(p), task_rq(p)->nr_running);
+
         set_tsk_need_resched(p);
 
         cpu = task_cpu(p);
@@ -3823,6 +3832,7 @@ need_resched_nonpreemptible:
                 hrtick_clear(rq);
 
         raw_spin_lock_irq(&rq->lock);
+        rq->dbg_in_schedule = 1;
         clear_tsk_need_resched(prev);
 
         switch_count = &prev->nivcsw;
@@ -3840,20 +3850,35 @@ need_resched_nonpreemptible:
                                 struct task_struct *to_wakeup;
 
                                 to_wakeup = wq_worker_sleeping(prev, cpu);
-                                if (to_wakeup)
+                                if (to_wakeup) {
+                                        TRACE_TASK(prev, "try_to_wake_up_local(%s/%d)\n", to_wakeup->comm, to_wakeup->pid);
                                         try_to_wake_up_local(to_wakeup);
+                                }
                         }
                         deactivate_task(rq, prev, DEQUEUE_SLEEP);
                 }
                 switch_count = &prev->nvcsw;
         }
 
+        if (test_tsk_need_resched(prev))
+                TRACE_TASK(prev, "need_resched before pre_schedule()\n");
+
         pre_schedule(rq, prev);
 
-        if (unlikely(!rq->nr_running))
+        if (test_tsk_need_resched(prev))
+                TRACE_TASK(prev, "need_resched after pre_schedule()\n");
+
+        if (unlikely(!rq->nr_running)) {
                 idle_balance(cpu, rq);
+                if (test_tsk_need_resched(prev))
+                        TRACE_TASK(prev, "need_resched after idle_balance\n");
+        }
 
         put_prev_task(rq, prev);
+
+        if (test_tsk_need_resched(prev))
+                TRACE_TASK(prev, "need_resched after put_prev\n");
+
         next = pick_next_task(rq);
 
         if (likely(prev != next)) {
@@ -3866,6 +3891,7 @@ need_resched_nonpreemptible:
 
                 TS_SCHED_END(next);
                 TS_CXS_START(next);
+                rq->dbg_in_schedule = 0;
                 context_switch(rq, prev, next); /* unlocks the rq */
                 TS_CXS_END(current);
                 /*
@@ -3878,6 +3904,7 @@ need_resched_nonpreemptible:
                 rq = cpu_rq(cpu);
         } else {
                 TS_SCHED_END(prev);
+                rq->dbg_in_schedule = 0;
                 raw_spin_unlock_irq(&rq->lock);
         }