return 0;
 }
 
-void kvm_update_cpuid(struct kvm_vcpu *vcpu)
+void kvm_update_cpuid_runtime(struct kvm_vcpu *vcpu)
 {
        struct kvm_cpuid_entry2 *best;
-       struct kvm_lapic *apic = vcpu->arch.apic;
 
        best = kvm_find_cpuid_entry(vcpu, 1, 0);
        if (best) {
                           vcpu->arch.apic_base & MSR_IA32_APICBASE_ENABLE);
        }
 
-       if (best && apic) {
-               if (cpuid_entry_has(best, X86_FEATURE_TSC_DEADLINE_TIMER))
-                       apic->lapic_timer.timer_mode_mask = 3 << 17;
-               else
-                       apic->lapic_timer.timer_mode_mask = 1 << 17;
-
-               kvm_apic_set_version(vcpu);
-       }
-
        best = kvm_find_cpuid_entry(vcpu, 7, 0);
        if (best && boot_cpu_has(X86_FEATURE_PKU) && best->function == 0x7)
                cpuid_entry_change(best, X86_FEATURE_OSPKE,
                                   kvm_read_cr4_bits(vcpu, X86_CR4_PKE));
 
        best = kvm_find_cpuid_entry(vcpu, 0xD, 0);
-       if (!best) {
-               vcpu->arch.guest_supported_xcr0 = 0;
-       } else {
-               vcpu->arch.guest_supported_xcr0 =
-                       (best->eax | ((u64)best->edx << 32)) & supported_xcr0;
+       if (best)
                best->ebx = xstate_required_size(vcpu->arch.xcr0, false);
-       }
 
        best = kvm_find_cpuid_entry(vcpu, 0xD, 1);
        if (best && (cpuid_entry_has(best, X86_FEATURE_XSAVES) ||
                                           vcpu->arch.ia32_misc_enable_msr &
                                           MSR_IA32_MISC_ENABLE_MWAIT);
        }
+}
+
+static void kvm_update_cpuid(struct kvm_vcpu *vcpu)
+{
+       struct kvm_lapic *apic = vcpu->arch.apic;
+       struct kvm_cpuid_entry2 *best;
+
+       best = kvm_find_cpuid_entry(vcpu, 1, 0);
+       if (best && apic) {
+               if (cpuid_entry_has(best, X86_FEATURE_TSC_DEADLINE_TIMER))
+                       apic->lapic_timer.timer_mode_mask = 3 << 17;
+               else
+                       apic->lapic_timer.timer_mode_mask = 1 << 17;
+
+               kvm_apic_set_version(vcpu);
+       }
+
+       best = kvm_find_cpuid_entry(vcpu, 0xD, 0);
+       if (!best)
+               vcpu->arch.guest_supported_xcr0 = 0;
+       else
+               vcpu->arch.guest_supported_xcr0 =
+                       (best->eax | ((u64)best->edx << 32)) & supported_xcr0;
 
        /* Note, maxphyaddr must be updated before tdp_level. */
        vcpu->arch.maxphyaddr = cpuid_query_maxphyaddr(vcpu);
 
        cpuid_fix_nx_cap(vcpu);
        kvm_x86_ops.cpuid_update(vcpu);
+       kvm_update_cpuid_runtime(vcpu);
        kvm_update_cpuid(vcpu);
 
        kvfree(cpuid_entries);
        }
 
        kvm_x86_ops.cpuid_update(vcpu);
+       kvm_update_cpuid_runtime(vcpu);
        kvm_update_cpuid(vcpu);
 out:
        return r;
 
        vcpu->arch.xcr0 = xcr0;
 
        if ((xcr0 ^ old_xcr0) & XFEATURE_MASK_EXTEND)
-               kvm_update_cpuid(vcpu);
+               kvm_update_cpuid_runtime(vcpu);
        return 0;
 }
 
                kvm_mmu_reset_context(vcpu);
 
        if ((cr4 ^ old_cr4) & (X86_CR4_OSXSAVE | X86_CR4_PKE))
-               kvm_update_cpuid(vcpu);
+               kvm_update_cpuid_runtime(vcpu);
 
        return 0;
 }
                        if (!guest_cpuid_has(vcpu, X86_FEATURE_XMM3))
                                return 1;
                        vcpu->arch.ia32_misc_enable_msr = data;
-                       kvm_update_cpuid(vcpu);
+                       kvm_update_cpuid_runtime(vcpu);
                } else {
                        vcpu->arch.ia32_misc_enable_msr = data;
                }
                kvm_x86_ops.set_efer(vcpu, 0);
 #endif
 
-       kvm_update_cpuid(vcpu);
+       kvm_update_cpuid_runtime(vcpu);
        kvm_mmu_reset_context(vcpu);
 }
 
                                (X86_CR4_OSXSAVE | X86_CR4_PKE));
        kvm_x86_ops.set_cr4(vcpu, sregs->cr4);
        if (cpuid_update_needed)
-               kvm_update_cpuid(vcpu);
+               kvm_update_cpuid_runtime(vcpu);
 
        idx = srcu_read_lock(&vcpu->kvm->srcu);
        if (is_pae_paging(vcpu)) {