1 files changed, 10 insertions, 11 deletions

diff --git a/litmus/sched_mc.c b/litmus/sched_mc.c
index 3d311505f437..53c24ac2172c 100644
--- a/litmus/sched_mc.c
+++ b/litmus/sched_mc.c
@@ -16,6 +16,7 @@
 #include <linux/pid.h>
 
 #include <litmus/litmus.h>
+#include <litmus/trace.h>
 #include <litmus/jobs.h>
 #include <litmus/sched_plugin.h>
 #include <litmus/edf_common.h>
@@ -534,8 +535,6 @@ static void remove_from_all(struct task_struct* task)
         struct crit_entry *ce;
         struct domain *dom = get_task_domain(task);
 
-        QT_START;
-
         TRACE_MC_TASK(task, "Removing from everything\n");
         BUG_ON(!task);
 
@@ -572,7 +571,6 @@ static void remove_from_all(struct task_struct* task)
 
         BUG_ON(is_queued(task));
         raw_spin_unlock(dom->lock);
-        QT_END;
 }
 
 /**
@@ -670,7 +668,6 @@ static lt_t __ce_timer_function(struct ce_dom_data *ce_data)
         struct domain *dom = ce->domain;
         struct task_struct *old_link = NULL;
         lt_t next_timer_abs;
-        QT_START;
 
         TRACE("MC level-A timer callback for CPU %d\n", ce_data->cpu);
 
@@ -700,7 +697,6 @@ static lt_t __ce_timer_function(struct ce_dom_data *ce_data)
                 raw_spin_unlock(dom->lock);
                 check_for_preempt(dom);
         }
-        QT_END;
         return next_timer_abs;
 }
 
@@ -712,10 +708,14 @@ static void ce_timer_function(struct rt_event *e)
         unsigned long flags;
         lt_t next_timer_abs;
 
+        TS_LVLA_RELEASE_START;
+
         local_irq_save(flags);
         next_timer_abs = __ce_timer_function(ce_data);
         add_event(per_cpu(cpus, ce_data->cpu).event_group, e, next_timer_abs);
         local_irq_restore(flags);
+
+        TS_LVLA_RELEASE_END;
 }
 #else /* else to CONFIG_MERGE_TIMERS */
 static enum hrtimer_restart ce_timer_function(struct hrtimer *timer)
@@ -725,10 +725,15 @@ static enum hrtimer_restart ce_timer_function(struct hrtimer *timer)
         unsigned long flags;
         lt_t next_timer_abs;
 
+        TS_LVLA_RELEASE_START;
+
         local_irq_save(flags);
         next_timer_abs = __ce_timer_function(ce_data);
         hrtimer_set_expires(timer, ns_to_ktime(next_timer_abs));
         local_irq_restore(flags);
+
+        TS_LVLA_RELEASE_END;
+
         return HRTIMER_RESTART;
 }
 #endif /* CONFIG_MERGE_TIMERS */
@@ -743,15 +748,11 @@ static void mc_release_jobs(rt_domain_t* rt, struct bheap* tasks)
         struct task_struct *first = bheap_peek(rt->order, tasks)->value;
         struct domain *dom = get_task_domain(first);
 
-        QT_START;
-
         raw_spin_lock_irqsave(dom->lock, flags);
         TRACE(TS "Jobs released\n", TA(first));
         __merge_ready(rt, tasks);
         check_for_preempt(dom);
         raw_spin_unlock_irqrestore(dom->lock, flags);
-
-        QT_END;
 }
 
 /**
@@ -890,7 +891,6 @@ static struct task_struct* mc_schedule(struct task_struct* prev)
         struct cpu_entry* entry = &__get_cpu_var(cpus);
         int i, out_of_time, sleep, preempt, exists, blocks, global, lower;
         struct task_struct *dtask = NULL, *ready_task = NULL, *next = NULL;
-        QT_START;
 
         local_irq_save(flags);
 
@@ -1004,7 +1004,6 @@ static struct task_struct* mc_schedule(struct task_struct* prev)
                 TRACE_MC_TASK(next, "Picked this task\n");
         } else if (exists && !next)
                 TRACE_ENTRY(entry, "Becomes idle at %llu\n", litmus_clock());
-        QT_END;
         return next;
 }