As the name of the field indicates, host_fpsimd_state is strictly
a host piece of data, and we reset this pointer on each PID change.
So let's move it where it belongs, and set it at load-time. Although
this is slightly more often, it is a well defined life-cycle which
matches other pieces of data.
Reviewed-by: Mark Brown <broonie@kernel.org>
Signed-off-by: Marc Zyngier <maz@kernel.org>
  */
 struct kvm_host_data {
        struct kvm_cpu_context host_ctxt;
+       struct user_fpsimd_state *fpsimd_state; /* hyp VA */
 
        /*
         * host_debug_state contains the host registers which are
        struct kvm_guest_debug_arch vcpu_debug_state;
        struct kvm_guest_debug_arch external_debug_state;
 
-       struct user_fpsimd_state *host_fpsimd_state;    /* hyp VA */
        struct task_struct *parent_task;
 
        /* VGIC state */
 
        if (ret)
                return ret;
 
-       vcpu->arch.host_fpsimd_state = kern_hyp_va(fpsimd);
-
        /*
         * We need to keep current's task_struct pinned until its data has been
         * unshared with the hypervisor to make sure it is not re-used by the
         * FP_STATE_FREE if the flag set.
         */
        vcpu->arch.fp_state = FP_STATE_HOST_OWNED;
+       *host_data_ptr(fpsimd_state) = kern_hyp_va(¤t->thread.uw.fpsimd_state);
 
        vcpu_clear_flag(vcpu, HOST_SVE_ENABLED);
        if (read_sysreg(cpacr_el1) & CPACR_EL1_ZEN_EL0EN)
 
 
        /* Write out the host state if it's in the registers */
        if (vcpu->arch.fp_state == FP_STATE_HOST_OWNED)
-               __fpsimd_save_state(vcpu->arch.host_fpsimd_state);
+               __fpsimd_save_state(*host_data_ptr(fpsimd_state));
 
        /* Restore the guest state */
        if (sve_guest)
 
        hyp_vcpu->vcpu.arch.fp_state    = host_vcpu->arch.fp_state;
 
        hyp_vcpu->vcpu.arch.debug_ptr   = kern_hyp_va(host_vcpu->arch.debug_ptr);
-       hyp_vcpu->vcpu.arch.host_fpsimd_state = host_vcpu->arch.host_fpsimd_state;
 
        hyp_vcpu->vcpu.arch.vsesr_el2   = host_vcpu->arch.vsesr_el2;