1 files changed, 8 insertions, 3 deletions

diff --git a/litmus/sched_gsn_edf.c b/litmus/sched_gsn_edf.c
index a223e69f2efb..5386f0a24587 100644
--- a/litmus/sched_gsn_edf.c
+++ b/litmus/sched_gsn_edf.c
@@ -18,6 +18,8 @@
 #include <litmus/edf_common.h>
 #include <litmus/sched_trace.h>
 
+#include <trace/events/sched.h>
+
 #include <litmus/heap.h>
 
 #include <linux/module.h>
@@ -321,7 +323,8 @@ static noinline void job_completion(struct task_struct *t, int forced)
 {
         BUG_ON(!t);
 
-        sched_trace_task_completion(t, forced);
+//      sched_trace_task_completion(t, forced);
+        trace_litmus_task_completion(t, forced);
 
         TRACE_TASK(t, "job_completion().\n");
 
@@ -330,7 +333,8 @@ static noinline void job_completion(struct task_struct *t, int forced)
         /* prepare for next period */
         prepare_for_next_period(t);
         if (is_released(t, litmus_clock()))
-                sched_trace_task_release(t);
+                trace_litmus_task_release(t);
+//              sched_trace_task_release(t);
         /* unlink */
         unlink(t);
         /* requeue
@@ -558,7 +562,8 @@ static void gsnedf_task_wake_up(struct task_struct *task)
                 if (is_tardy(task, now)) {
                         /* new sporadic release */
                         release_at(task, now);
-                        sched_trace_task_release(task);
+//                      sched_trace_task_release(task);
+                        trace_litmus_task_release(task);
                 }
                 else {
                         if (task->rt.time_slice) {