1 files changed, 4 insertions, 0 deletions

diff --git a/kernel/softirq.c b/kernel/softirq.c
index be4b8fab3637..ae77c5c1d17e 100644
--- a/kernel/softirq.c
+++ b/kernel/softirq.c
@@ -35,6 +35,7 @@
 
 #ifdef CONFIG_LITMUS_NVIDIA
 #include <litmus/nvidia_info.h>
+#include <litmus/trace.h>
 #endif
 
 /*
@@ -441,6 +442,9 @@ void __tasklet_schedule(struct tasklet_struct *t)
                                 if(likely(_litmus_tasklet_schedule(t,nvidia_device)))
                                 {
                                         unlock_nv_registry(nvidia_device, &flags);
+
+                                        TS_NV_RELEASE_BOTISR_END;
+
                                         return;
                                 }
                                 else