* struct vic_device - VIC PM device
  * @irq: The IRQ number for the base of the VIC.
  * @base: The register base for the VIC.
+ * @valid_sources: A bitmask of valid interrupts
  * @resume_sources: A bitmask of interrupts for resume.
  * @resume_irqs: The IRQs enabled for resume.
  * @int_select: Save for VIC_INT_SELECT.
 struct vic_device {
        void __iomem    *base;
        int             irq;
+       u32             valid_sources;
        u32             resume_sources;
        u32             resume_irqs;
        u32             int_select;
 late_initcall(vic_pm_init);
 #endif /* CONFIG_PM */
 
+static struct irq_chip vic_chip;
+
+static int vic_irqdomain_map(struct irq_domain *d, unsigned int irq,
+                            irq_hw_number_t hwirq)
+{
+       struct vic_device *v = d->host_data;
+
+       /* Skip invalid IRQs, only register handlers for the real ones */
+       if (!(v->valid_sources & (1 << hwirq)))
+               return -ENOTSUPP;
+       irq_set_chip_and_handler(irq, &vic_chip, handle_level_irq);
+       irq_set_chip_data(irq, v->base);
+       set_irq_flags(irq, IRQF_VALID | IRQF_PROBE);
+       return 0;
+}
+
+static struct irq_domain_ops vic_irqdomain_ops = {
+       .map = vic_irqdomain_map,
+       .xlate = irq_domain_xlate_onetwocell,
+};
+
 /**
  * vic_register() - Register a VIC.
  * @base: The base address of the VIC.
 
        v = &vic_devices[vic_id];
        v->base = base;
+       v->valid_sources = valid_sources;
        v->resume_sources = resume_sources;
        v->irq = irq;
        vic_id++;
        }
 }
 
-static void __init vic_set_irq_sources(void __iomem *base,
-                               unsigned int irq_start, u32 vic_sources)
-{
-       unsigned int i;
-
-       for (i = 0; i < 32; i++) {
-               if (vic_sources & (1 << i)) {
-                       unsigned int irq = irq_start + i;
-
-                       irq_set_chip_and_handler(irq, &vic_chip,
-                                                handle_level_irq);
-                       irq_set_chip_data(irq, base);
-                       set_irq_flags(irq, IRQF_VALID | IRQF_PROBE);
-               }
-       }
-}
-
 /*
  * The PL190 cell from ARM has been modified by ST to handle 64 interrupts.
  * The original cell has 32 interrupts, while the modified one has 64,
                writel(32, base + VIC_PL190_DEF_VECT_ADDR);
        }
 
-       vic_set_irq_sources(base, irq_start, vic_sources);
        vic_register(base, irq_start, vic_sources, 0, node);
 }
 
 
        vic_init2(base);
 
-       vic_set_irq_sources(base, irq_start, vic_sources);
-
        vic_register(base, irq_start, vic_sources, resume_sources, node);
 }