kfree(tmp);
                return err;
        } else {
+               int ret;
+
                /*
                 * For 64-bit frames and 32-bit fsave frames, restore the user
                 * state to the registers directly (with exceptions handled).
                 */
-               if (copy_user_to_fpregs_zeroing(buf_fx, xfeatures, fx_only)) {
+               if (use_xsave()) {
+                       if ((unsigned long)buf_fx % 64 || fx_only) {
+                               u64 init_bv = xfeatures_mask & ~XFEATURE_MASK_FPSSE;
+                               copy_kernel_to_xregs(&init_fpstate.xsave, init_bv);
+                               ret = copy_user_to_fxregs(buf_fx);
+                       } else {
+                               u64 init_bv = xfeatures_mask & ~xfeatures;
+                               if (unlikely(init_bv))
+                                       copy_kernel_to_xregs(&init_fpstate.xsave, init_bv);
+                               ret = copy_user_to_xregs(buf_fx, xfeatures);
+                       }
+               } else if (use_fxsr()) {
+                       ret = copy_user_to_fxregs(buf_fx);
+               } else
+                       ret = copy_user_to_fregs(buf_fx);
+
+               if (ret) {
                        fpu__clear(fpu);
                        return -1;
                }