}
 }
 
+static bool s1pie_enabled(struct kvm_vcpu *vcpu, enum trans_regime regime)
+{
+       if (!kvm_has_feat(vcpu->kvm, ID_AA64MMFR3_EL1, S1PIE, IMP))
+               return false;
+
+       switch (regime) {
+       case TR_EL2:
+       case TR_EL20:
+               return vcpu_read_sys_reg(vcpu, TCR2_EL2) & TCR2_EL2_PIE;
+       case TR_EL10:
+               return  (__vcpu_sys_reg(vcpu, HCRX_EL2) & HCRX_EL2_TCR2En) &&
+                       (__vcpu_sys_reg(vcpu, TCR2_EL1) & TCR2_EL1x_PIE);
+       default:
+               BUG();
+       }
+}
+
 static int setup_s1_walk(struct kvm_vcpu *vcpu, u32 op, struct s1_walk_info *wi,
                         struct s1_walk_result *wr, u64 va)
 {
                    (va55 ?
                     FIELD_GET(TCR_HPD1, tcr) :
                     FIELD_GET(TCR_HPD0, tcr)));
+       /* R_JHSVW */
+       wi->hpd |= s1pie_enabled(vcpu, wi->regime);
 
        /* Someone was silly enough to encode TG0/TG1 differently */
        if (va55) {