1 files changed, 9 insertions, 3 deletions

diff --git a/arch/x86/kvm/x86.c b/arch/x86/kvm/x86.c
index c4d451ed1573..19047eafa38d 100644
--- a/arch/x86/kvm/x86.c
+++ b/arch/x86/kvm/x86.c
@@ -6016,7 +6016,9 @@ int kvm_arch_vcpu_setup(struct kvm_vcpu *vcpu)
         int r;
 
         vcpu->arch.mtrr_state.have_fixed = 1;
-        vcpu_load(vcpu);
+        r = vcpu_load(vcpu);
+        if (r)
+                return r;
         r = kvm_arch_vcpu_reset(vcpu);
         if (r == 0)
                 r = kvm_mmu_setup(vcpu);
@@ -6027,9 +6029,11 @@ int kvm_arch_vcpu_setup(struct kvm_vcpu *vcpu)
 
 void kvm_arch_vcpu_destroy(struct kvm_vcpu *vcpu)
 {
+        int r;
         vcpu->arch.apf.msr_val = 0;
 
-        vcpu_load(vcpu);
+        r = vcpu_load(vcpu);
+        BUG_ON(r);
         kvm_mmu_unload(vcpu);
         vcpu_put(vcpu);
 
@@ -6275,7 +6279,9 @@ int kvm_arch_init_vm(struct kvm *kvm, unsigned long type)
 
 static void kvm_unload_vcpu_mmu(struct kvm_vcpu *vcpu)
 {
-        vcpu_load(vcpu);
+        int r;
+        r = vcpu_load(vcpu);
+        BUG_ON(r);
         kvm_mmu_unload(vcpu);
         vcpu_put(vcpu);
 }