1 files changed, 10 insertions, 5 deletions

diff --git a/litmus/sched_gsn_edf.c b/litmus/sched_gsn_edf.c
index b40ff7ba4f0e..30c745fe33a7 100644
--- a/litmus/sched_gsn_edf.c
+++ b/litmus/sched_gsn_edf.c
@@ -37,6 +37,7 @@
 
 #ifdef CONFIG_LITMUS_PAI_SOFTIRQD
 #include <linux/interrupt.h>
+#include <litmus/trace.h>
 #endif
 
 #ifdef CONFIG_LITMUS_NVIDIA
@@ -523,6 +524,9 @@ static void do_lit_tasklets(struct task_struct* sched_task)
         unsigned long flags;
         
         while(work_to_do) {
+                
+                TS_NV_SCHED_BOTISR_START;
+                
                 // remove tasklet at head of list if it has higher priority.
                 raw_spin_lock_irqsave(&gsnedf_lock, flags);     
                 
@@ -576,9 +580,10 @@ static void do_lit_tasklets(struct task_struct* sched_task)
                  TRACE("%s: done.\n", __FUNCTION__);
                  */
                 
-                
                 raw_spin_unlock_irqrestore(&gsnedf_lock, flags);
                 
+                TS_NV_SCHED_BOTISR_END;
+                
                 if(tasklet) {
                         __do_lit_tasklet(tasklet, 0ul);
                         tasklet = NULL; 
@@ -1069,6 +1074,10 @@ static void gsnedf_task_exit(struct task_struct * t)
 {
         unsigned long flags;
 
+#ifdef CONFIG_LITMUS_PAI_SOFTIRQD
+        flush_tasklets(t);
+#endif          
+        
         /* unlink if necessary */
         raw_spin_lock_irqsave(&gsnedf_lock, flags);
         unlink(t);
@@ -1077,10 +1086,6 @@ static void gsnedf_task_exit(struct task_struct * t)
                 tsk_rt(t)->scheduled_on = NO_CPU;
         }
         raw_spin_unlock_irqrestore(&gsnedf_lock, flags);
-
-#ifdef CONFIG_LITMUS_PAI_SOFTIRQD
-        flush_tasklets(t);
-#endif  
         
         BUG_ON(!is_realtime(t));
         TRACE_TASK(t, "RIP\n");