{
        struct irq_desc *desc = irq_to_desc(irq);
 
-       desc->irq_data.chip->mask(irq);
+       desc->irq_data.chip->irq_mask(&desc->irq_data);
        desc->status |= IRQ_MASKED;
 }
 
 /* Temporary migration helpers */
+static void compat_irq_mask(struct irq_data *data)
+{
+       data->chip->mask(data->irq);
+}
+
 static void compat_bus_lock(struct irq_data *data)
 {
        data->chip->bus_lock(data->irq);
                chip->irq_bus_lock = compat_bus_lock;
        if (chip->bus_sync_unlock)
                chip->irq_bus_sync_unlock = compat_bus_sync_unlock;
+
+       if (chip->mask)
+               chip->irq_mask = compat_irq_mask;
 }
 
 static inline void mask_ack_irq(struct irq_desc *desc, int irq)
        if (desc->irq_data.chip->mask_ack)
                desc->irq_data.chip->mask_ack(irq);
        else {
-               desc->irq_data.chip->mask(irq);
+               desc->irq_data.chip->irq_mask(&desc->irq_data);
                if (desc->irq_data.chip->ack)
                        desc->irq_data.chip->ack(irq);
        }
        desc->status |= IRQ_MASKED;
 }
 
-static inline void mask_irq(struct irq_desc *desc, int irq)
+static inline void mask_irq(struct irq_desc *desc)
 {
-       if (desc->irq_data.chip->mask) {
-               desc->irq_data.chip->mask(irq);
+       if (desc->irq_data.chip->irq_mask) {
+               desc->irq_data.chip->irq_mask(&desc->irq_data);
                desc->status |= IRQ_MASKED;
        }
 }
        action = desc->action;
        if (unlikely(!action || (desc->status & IRQ_DISABLED))) {
                desc->status |= IRQ_PENDING;
-               mask_irq(desc, irq);
+               mask_irq(desc);
                goto out;
        }
 
                irqreturn_t action_ret;
 
                if (unlikely(!action)) {
-                       mask_irq(desc, irq);
+                       mask_irq(desc);
                        goto out_unlock;
                }