return frstor_checking(&fpu->state->fsave);
 }
 
-static inline int restore_fpu_checking(struct task_struct *tsk)
+static inline int restore_fpu_checking(struct fpu *fpu)
 {
        /*
         * AMD K7/K8 CPUs don't save/restore FDP/FIP/FOP unless an exception is
                        "fnclex\n\t"
                        "emms\n\t"
                        "fildl %P[addr]"        /* set F?P to defined value */
-                       : : [addr] "m" (tsk->thread.fpu.has_fpu));
+                       : : [addr] "m" (fpu->has_fpu));
        }
 
-       return fpu_restore_checking(&tsk->thread.fpu);
+       return fpu_restore_checking(fpu);
 }
 
 /* Must be paired with an 'stts' after! */
  */
 static inline void switch_fpu_finish(struct task_struct *new, fpu_switch_t fpu)
 {
+       struct fpu *new_fpu = &new->thread.fpu;
+
        if (fpu.preload) {
-               if (unlikely(restore_fpu_checking(new)))
+               if (unlikely(restore_fpu_checking(new_fpu)))
                        fpu_reset_state(new);
        }
 }
 
        struct fpu *fpu = &me->thread.fpu;
 
        if (fpu->has_fpu) {
-               if (WARN_ON(restore_fpu_checking(me)))
+               if (WARN_ON(restore_fpu_checking(fpu)))
                        fpu_reset_state(me);
        } else if (!use_eager_fpu()) {
                stts();
        /* Avoid __kernel_fpu_begin() right after __thread_fpu_begin() */
        kernel_fpu_disable();
        __thread_fpu_begin(fpu);
-       if (unlikely(restore_fpu_checking(tsk))) {
+       if (unlikely(restore_fpu_checking(fpu))) {
                fpu_reset_state(tsk);
                force_sig_info(SIGSEGV, SEND_SIG_PRIV, tsk);
        } else {