gic_map_to_vpe(irq, mips_cm_vp_id(cpumask_first(&tmp)));
 
        /* Update the pcpu_masks */
-       for (i = 0; i < NR_CPUS; i++)
+       for (i = 0; i < gic_vpes; i++)
                clear_bit(irq, pcpu_masks[i].pcpu_mask);
        set_bit(irq, pcpu_masks[cpumask_first(&tmp)].pcpu_mask);
 
 
        gic_ipi_domain->bus_token = DOMAIN_BUS_IPI;
 
-       /* Make the last 2 * NR_CPUS available for IPIs */
-       bitmap_set(ipi_resrv, gic_shared_intrs - 2 * NR_CPUS, 2 * NR_CPUS);
+       /* Make the last 2 * gic_vpes available for IPIs */
+       bitmap_set(ipi_resrv, gic_shared_intrs - 2 * gic_vpes, 2 * gic_vpes);
 
        gic_basic_init();