#define KVM_IRQCHIP_NUM_PINS   (1020 - 32)
 
 #define irq_is_ppi(irq) ((irq) >= VGIC_NR_SGIS && (irq) < VGIC_NR_PRIVATE_IRQS)
+#define irq_is_spi(irq) ((irq) >= VGIC_NR_PRIVATE_IRQS && \
+                        (irq) <= VGIC_MAX_SPI)
 
 enum vgic_type {
        VGIC_V2,                /* Good ol' GICv2 */
 
        /*
         * A valid interrupt configuration for the PMU is either to have a
         * properly configured interrupt number and using an in-kernel
-        * irqchip, or to neither set an IRQ nor create an in-kernel irqchip.
+        * irqchip, or to not have an in-kernel GIC and not set an IRQ.
         */
-       if (kvm_arm_pmu_irq_initialized(vcpu) != irqchip_in_kernel(vcpu->kvm))
-               return -EINVAL;
+       if (irqchip_in_kernel(vcpu->kvm)) {
+               int irq = vcpu->arch.pmu.irq_num;
+               if (!kvm_arm_pmu_irq_initialized(vcpu))
+                       return -EINVAL;
+
+               /*
+                * If we are using an in-kernel vgic, at this point we know
+                * the vgic will be initialized, so we can check the PMU irq
+                * number against the dimensions of the vgic and make sure
+                * it's valid.
+                */
+               if (!irq_is_ppi(irq) && !vgic_valid_spi(vcpu->kvm, irq))
+                       return -EINVAL;
+       } else if (kvm_arm_pmu_irq_initialized(vcpu)) {
+                  return -EINVAL;
+       }
 
        kvm_pmu_vcpu_reset(vcpu);
        vcpu->arch.pmu.ready = true;
                        return -EFAULT;
 
                /* The PMU overflow interrupt can be a PPI or a valid SPI. */
-               if (!(irq_is_ppi(irq) || vgic_valid_spi(vcpu->kvm, irq)))
+               if (!(irq_is_ppi(irq) || irq_is_spi(irq)))
                        return -EINVAL;
 
                if (!pmu_irq_is_valid(vcpu->kvm, irq))