#include <asm/desc.h>
 #include <asm/mtrr.h>
 #include <asm/mce.h>
+#include <asm/i387.h>
 
 #define MAX_IO_MSRS 256
 #define CR0_RESERVED_BITS                                              \
 {
        unsigned after_mxcsr_mask;
 
-       /*
-        * Touch the fpu the first time in non atomic context as if
-        * this is the first fpu instruction the exception handler
-        * will fire before the instruction returns and it'll have to
-        * allocate ram with GFP_KERNEL.
-        */
-       if (!used_math())
-               kvm_fx_save(&vcpu->arch.host_fx_image);
-
        /* Initialize guest FPU by resetting ours and saving into guest's */
        preempt_disable();
-       kvm_fx_save(&vcpu->arch.host_fx_image);
        kvm_fx_finit();
        kvm_fx_save(&vcpu->arch.guest_fx_image);
-       kvm_fx_restore(&vcpu->arch.host_fx_image);
        preempt_enable();
 
        vcpu->arch.cr0 |= X86_CR0_ET;
                return;
 
        vcpu->guest_fpu_loaded = 1;
-       kvm_fx_save(&vcpu->arch.host_fx_image);
+       unlazy_fpu(current);
        kvm_fx_restore(&vcpu->arch.guest_fx_image);
        trace_kvm_fpu(1);
 }
 
        vcpu->guest_fpu_loaded = 0;
        kvm_fx_save(&vcpu->arch.guest_fx_image);
-       kvm_fx_restore(&vcpu->arch.host_fx_image);
        ++vcpu->stat.fpu_reload;
        set_bit(KVM_REQ_DEACTIVATE_FPU, &vcpu->requests);
        trace_kvm_fpu(0);
 {
        int r;
 
-       /* We do fxsave: this must be aligned. */
-       BUG_ON((unsigned long)&vcpu->arch.host_fx_image & 0xF);
-
        vcpu->arch.mtrr_state.have_fixed = 1;
        vcpu_load(vcpu);
        r = kvm_arch_vcpu_reset(vcpu);