1 files changed, 1 insertions, 4 deletions

diff --git a/arch/powerpc/kernel/traps.c b/arch/powerpc/kernel/traps.c
index 1a0141426cd..f19d9777d3c 100644
--- a/arch/powerpc/kernel/traps.c
+++ b/arch/powerpc/kernel/traps.c
@@ -1387,10 +1387,7 @@ void SPEFloatingPointException(struct pt_regs *regs)
         int code = 0;
         int err;
 
-        preempt_disable();
-        if (regs->msr & MSR_SPE)
-                giveup_spe(current);
-        preempt_enable();
+        flush_spe_to_thread(current);
 
         spefscr = current->thread.spefscr;
         fpexc_mode = current->thread.fpexc_mode;