Hookup sched_trace_XXX() tracing in Linux scheduler

This patch adds context switch tracing to the main Linux scheduler.

1 files changed, 6 insertions, 1 deletions

diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index 435d755cf9bf..1f8d205a8b7f 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -89,6 +89,7 @@
 #include "../smpboot.h"
 
 #include <litmus/trace.h>
+#include <litmus/sched_trace.h>
 
 #define CREATE_TRACE_POINTS
 #include <trace/events/sched.h>
@@ -2858,6 +2859,8 @@ asmlinkage __visible void schedule_tail(struct task_struct *prev)
 
         rq = finish_task_switch(prev);
 
+        sched_trace_task_switch_to(current);
+
         if (unlikely(sched_state_validate_switch()))
                 litmus_reschedule_local();
 
@@ -3352,13 +3355,14 @@ static void __sched notrace __schedule(bool preempt)
         int cpu;
 
         TS_SCHED_START;
-
         sched_state_entered_schedule();
 
         cpu = smp_processor_id();
         rq = cpu_rq(cpu);
         prev = rq->curr;
 
+        sched_trace_task_switch_away(prev);
+
         schedule_debug(prev);
 
         if (sched_feat(HRTICK))
@@ -3431,6 +3435,7 @@ static void __sched notrace __schedule(bool preempt)
         }
 
         TS_SCHED2_START(prev);
+        sched_trace_task_switch_to(current);
 
         if (unlikely(sched_state_validate_switch()))
                 litmus_reschedule_local();