Hook into __schedule() to set litmus_preemption_in_progress

1 files changed, 4 insertions, 0 deletions

diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index f2f3a3000a1c..435d755cf9bf 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -3378,6 +3378,8 @@ static void __sched notrace __schedule(bool preempt)
 
         rq->clock_skip_update <<= 1; /* promote REQ to ACT */
 
+        this_cpu_write(litmus_preemption_in_progress, preempt);
+
         switch_count = &prev->nivcsw;
         if (!preempt && prev->state) {
                 if (unlikely(signal_pending_state(prev->state, prev))) {
@@ -3410,6 +3412,8 @@ static void __sched notrace __schedule(bool preempt)
         clear_preempt_need_resched();
         rq->clock_skip_update = 0;
 
+        this_cpu_write(litmus_preemption_in_progress, false);
+
         if (likely(prev != next)) {
                 rq->nr_switches++;
                 rq->curr = next;