1 files changed, 2 insertions, 0 deletions

diff --git a/arch/ia64/kvm/kvm-ia64.c b/arch/ia64/kvm/kvm-ia64.c
index 70d224d4264c..8213efe1998c 100644
--- a/arch/ia64/kvm/kvm-ia64.c
+++ b/arch/ia64/kvm/kvm-ia64.c
@@ -662,6 +662,7 @@ again:
                 goto vcpu_run_fail;
 
         srcu_read_unlock(&vcpu->kvm->srcu, idx);
+        vcpu->mode = IN_GUEST_MODE;
         kvm_guest_enter();
 
         /*
@@ -683,6 +684,7 @@ again:
          */
         barrier();
         kvm_guest_exit();
+        vcpu->mode = OUTSIDE_GUEST_MODE;
         preempt_enable();
 
         idx = srcu_read_lock(&vcpu->kvm->srcu);