1 files changed, 2 insertions, 0 deletions

diff --git a/include/linux/sched.h b/include/linux/sched.h
index 3e6e669db92b..ade13badfacc 100644
--- a/include/linux/sched.h
+++ b/include/linux/sched.h
@@ -126,6 +126,7 @@ struct sched_attr {
 };
 
 #include <litmus/rt_param.h>
+#include <litmus/preempt.h>
 
 struct futex_pi_state;
 struct robust_list_head;
@@ -2784,6 +2785,7 @@ static inline int test_tsk_thread_flag(struct task_struct *tsk, int flag)
 static inline void set_tsk_need_resched(struct task_struct *tsk)
 {
         set_tsk_thread_flag(tsk,TIF_NEED_RESCHED);
+        sched_state_will_schedule(tsk);
 }
 
 static inline void clear_tsk_need_resched(struct task_struct *tsk)