1 files changed, 8 insertions, 0 deletions

diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index 50f9f4c45ce9..377273d6cecc 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -2294,6 +2294,10 @@ asmlinkage __visible void schedule_tail(struct task_struct *prev)
         preempt_disable();
         rq = finish_task_switch(prev);
         post_schedule(rq);
+
+        if (sched_state_validate_switch())
+                litmus_reschedule_local();
+
         preempt_enable();
 
         if (current->set_child_tid)
@@ -2745,6 +2749,7 @@ static void __sched __schedule(void)
         int cpu;
 
         preempt_disable();
+        sched_state_entered_schedule();
         cpu = smp_processor_id();
         rq = cpu_rq(cpu);
         rcu_note_context_switch();
@@ -2819,6 +2824,9 @@ static void __sched __schedule(void)
 
         post_schedule(rq);
 
+        if (unlikely(sched_state_validate_switch()))
+                litmus_reschedule_local();
+
         sched_preempt_enable_no_resched();
 
         TS_SCHED2_END(prev);