1 files changed, 3 insertions, 0 deletions

diff --git a/arch/mips/kernel/entry.S b/arch/mips/kernel/entry.S
index 8d83fc2a96b7..38a302919e6b 100644
--- a/arch/mips/kernel/entry.S
+++ b/arch/mips/kernel/entry.S
@@ -11,6 +11,7 @@
 #include <asm/asm.h>
 #include <asm/asmmacro.h>
 #include <asm/compiler.h>
+#include <asm/irqflags.h>
 #include <asm/regdef.h>
 #include <asm/mipsregs.h>
 #include <asm/stackframe.h>
@@ -119,6 +120,7 @@ work_pending:
         andi    t0, a2, _TIF_NEED_RESCHED # a2 is preloaded with TI_FLAGS
         beqz    t0, work_notifysig
 work_resched:
+        TRACE_IRQS_OFF
         jal     schedule
 
         local_irq_disable               # make sure need_resched and
@@ -155,6 +157,7 @@ syscall_exit_work:
         beqz    t0, work_pending        # trace bit set?
         local_irq_enable                # could let syscall_trace_leave()
                                         # call schedule() instead
+        TRACE_IRQS_ON
         move    a0, sp
         jal     syscall_trace_leave
         b       resume_userspace