1 files changed, 15 insertions, 0 deletions

diff --git a/arch/mips/kernel/entry.S b/arch/mips/kernel/entry.S
index 01e7fa86aa43..766655f35250 100644
--- a/arch/mips/kernel/entry.S
+++ b/arch/mips/kernel/entry.S
@@ -113,6 +113,21 @@ FEXPORT(restore_all)			# restore full frame
         RESTORE_AT
         RESTORE_STATIC
 FEXPORT(restore_partial)                # restore partial frame
+#ifdef CONFIG_TRACE_IRQFLAGS
+        SAVE_STATIC
+        SAVE_AT
+        SAVE_TEMP
+        LONG_L  v0, PT_STATUS(sp)
+        and     v0, 1
+        beqz    v0, 1f
+        jal     trace_hardirqs_on
+        b       2f
+1:      jal     trace_hardirqs_off
+2:
+        RESTORE_TEMP
+        RESTORE_AT
+        RESTORE_STATIC
+#endif
         RESTORE_SOME
         RESTORE_SP_AND_RET
         .set    at