1 files changed, 14 insertions, 0 deletions

diff --git a/arch/x86/kernel/irq.c b/arch/x86/kernel/irq.c
index 91fd0c70a18a..50abbc6b7429 100644
--- a/arch/x86/kernel/irq.c
+++ b/arch/x86/kernel/irq.c
@@ -8,6 +8,10 @@
 #include <linux/smp.h>
 #include <linux/ftrace.h>
 
+#ifdef CONFIG_LITMUS_NVIDIA
+#include <litmus/sched_trace.h>
+#endif
+
 #include <asm/apic.h>
 #include <asm/io_apic.h>
 #include <asm/irq.h>
@@ -244,7 +248,17 @@ unsigned int __irq_entry do_IRQ(struct pt_regs *regs)
                                 __func__, smp_processor_id(), vector, irq);
         }
 
+//#ifndef CONFIG_LITMUS_NVIDIA
         irq_exit();
+//#else
+        /* skip softirqs if we're tracing an interrupt top-half */
+        /* comment out if-statement if we want to trace with bh on. */
+        //if(!is_interrupt_tracing_active())
+//      irq_exit();
+
+
+//      sched_trace_nv_interrupt_end();
+//#endif
 
         set_irq_regs(old_regs);
         return 1;