1 files changed, 13 insertions, 2 deletions

diff --git a/arch/sparc/include/asm/ttable.h b/arch/sparc/include/asm/ttable.h
index 5708ba2719fb..bb2c0770a6ac 100644
--- a/arch/sparc/include/asm/ttable.h
+++ b/arch/sparc/include/asm/ttable.h
@@ -2,6 +2,7 @@
 #define _SPARC64_TTABLE_H
 
 #include <asm/utrap.h>
+#include <asm/pil.h>
 
 #ifdef __ASSEMBLY__
 #include <asm/thread_info.h>
@@ -123,7 +124,7 @@
 
 #define TRAP_IRQ(routine, level)                        \
         rdpr    %pil, %g2;                              \
-        wrpr    %g0, 15, %pil;                          \
+        wrpr    %g0, PIL_NORMAL_MAX, %pil;              \
         sethi   %hi(1f-4), %g7;                         \
         ba,pt   %xcc, etrap_irq;                        \
          or     %g7, %lo(1f-4), %g7;                    \
@@ -143,7 +144,7 @@
 
 #define TRAP_IRQ(routine, level)                        \
         rdpr    %pil, %g2;                              \
-        wrpr    %g0, 15, %pil;                          \
+        wrpr    %g0, PIL_NORMAL_MAX, %pil;              \
         ba,pt   %xcc, etrap_irq;                        \
          rd     %pc, %g7;                               \
         mov     level, %o0;                             \
@@ -153,6 +154,16 @@
 
 #endif
 
+#define TRAP_NMI_IRQ(routine, level)                    \
+        rdpr    %pil, %g2;                              \
+        wrpr    %g0, PIL_NMI, %pil;                     \
+        ba,pt   %xcc, etrap_irq;                        \
+         rd     %pc, %g7;                               \
+        mov     level, %o0;                             \
+        call    routine;                                \
+         add    %sp, PTREGS_OFF, %o1;                   \
+        ba,a,pt %xcc, rtrap_irq;
+
 #define TRAP_IVEC TRAP_NOSAVE(do_ivec)
 
 #define BTRAP(lvl) TRAP_ARG(bad_trap, lvl)