1 files changed, 6 insertions, 0 deletions

diff --git a/arch/mips/kernel/genex.S b/arch/mips/kernel/genex.S
index b563811b1b27..37fda3dcdfc5 100644
--- a/arch/mips/kernel/genex.S
+++ b/arch/mips/kernel/genex.S
@@ -13,6 +13,7 @@
 #include <asm/asm.h>
 #include <asm/asmmacro.h>
 #include <asm/cacheops.h>
+#include <asm/irqflags.h>
 #include <asm/regdef.h>
 #include <asm/fpregdef.h>
 #include <asm/mipsregs.h>
@@ -128,6 +129,7 @@ handle_vcei:
 NESTED(handle_int, PT_SIZE, sp)
         SAVE_ALL
         CLI
+        TRACE_IRQS_OFF
 
         PTR_LA  ra, ret_from_irq
         move    a0, sp
@@ -216,6 +218,7 @@ NESTED(except_vec_vi_handler, 0, sp)
         _ehb
 #endif /* CONFIG_MIPS_MT_SMTC */
         CLI
+        TRACE_IRQS_OFF
         move    a0, sp
         jalr    v0
         j       ret_from_irq
@@ -288,11 +291,13 @@ NESTED(nmi_handler, PT_SIZE, sp)
         .endm
 
         .macro  __build_clear_sti
+        TRACE_IRQS_ON
         STI
         .endm
 
         .macro  __build_clear_cli
         CLI
+        TRACE_IRQS_OFF
         .endm
 
         .macro  __build_clear_fpe
@@ -300,6 +305,7 @@ NESTED(nmi_handler, PT_SIZE, sp)
         li      a2, ~(0x3f << 12)
         and     a2, a1
         ctc1    a2, fcr31
+        TRACE_IRQS_ON
         STI
         .endm