Litmus core: trace syscall overhead in sched_yield()

1 files changed, 3 insertions, 0 deletions

diff --git a/litmus/sched_litmus.c b/litmus/sched_litmus.c
index 231c86daa1f9..0687be0c8a78 100644
--- a/litmus/sched_litmus.c
+++ b/litmus/sched_litmus.c
@@ -196,6 +196,7 @@ static void yield_task_litmus(struct rq *rq)
 {
         TS_SYSCALL_IN_START;
 
+        TS_SYSCALL_OUT_END;
         BUG_ON(rq->curr != current);
         /* sched_yield() is called to trigger delayed preemptions.
          * Thus, mark the current task as needing to be rescheduled.
@@ -204,6 +205,8 @@ static void yield_task_litmus(struct rq *rq)
          */
         clear_exit_np(current);
         litmus_reschedule_local();
+
+        TS_SYSCALL_OUT_START;
 }
 
 /* Plugins are responsible for this.