return &its_dev->event_map.vlpi_maps[event];
 }
 
-static struct its_collection *irq_to_col(struct irq_data *d)
+static struct its_vlpi_map *get_vlpi_map(struct irq_data *d)
+{
+       if (irqd_is_forwarded_to_vcpu(d)) {
+               struct its_device *its_dev = irq_data_get_irq_chip_data(d);
+               u32 event = its_get_event_id(d);
+
+               return dev_event_to_vlpi_map(its_dev, event);
+       }
+
+       return NULL;
+}
+
+static int irq_to_cpuid(struct irq_data *d)
 {
        struct its_device *its_dev = irq_data_get_irq_chip_data(d);
+       struct its_vlpi_map *map = get_vlpi_map(d);
+
+       if (map)
+               return map->vpe->col_idx;
 
-       return dev_event_to_col(its_dev, its_get_event_id(d));
+       return its_dev->event_map.col_map[its_get_event_id(d)];
 }
 
 static struct its_collection *valid_col(struct its_collection *col)
 /*
  * irqchip functions - assumes MSI, mostly.
  */
-static struct its_vlpi_map *get_vlpi_map(struct irq_data *d)
-{
-       if (irqd_is_forwarded_to_vcpu(d)) {
-               struct its_device *its_dev = irq_data_get_irq_chip_data(d);
-               u32 event = its_get_event_id(d);
-
-               return dev_event_to_vlpi_map(its_dev, event);
-       }
-
-       return NULL;
-}
-
 static void lpi_write_config(struct irq_data *d, u8 clr, u8 set)
 {
        struct its_vlpi_map *map = get_vlpi_map(d);
 
 static void direct_lpi_inv(struct irq_data *d)
 {
-       struct its_collection *col;
+       struct its_vlpi_map *map = get_vlpi_map(d);
        void __iomem *rdbase;
+       u64 val;
+
+       if (map) {
+               struct its_device *its_dev = irq_data_get_irq_chip_data(d);
+
+               WARN_ON(!is_v4_1(its_dev->its));
+
+               val  = GICR_INVLPIR_V;
+               val |= FIELD_PREP(GICR_INVLPIR_VPEID, map->vpe->vpe_id);
+               val |= FIELD_PREP(GICR_INVLPIR_INTID, map->vintid);
+       } else {
+               val = d->hwirq;
+       }
 
        /* Target the redistributor this LPI is currently routed to */
-       col = irq_to_col(d);
-       rdbase = per_cpu_ptr(gic_rdists->rdist, col->col_id)->rd_base;
-       gic_write_lpir(d->hwirq, rdbase + GICR_INVLPIR);
+       rdbase = per_cpu_ptr(gic_rdists->rdist, irq_to_cpuid(d))->rd_base;
+       gic_write_lpir(val, rdbase + GICR_INVLPIR);
 
        wait_for_syncr(rdbase);
 }
        struct its_device *its_dev = irq_data_get_irq_chip_data(d);
 
        lpi_write_config(d, clr, set);
-       if (gic_rdists->has_direct_lpi && !irqd_is_forwarded_to_vcpu(d))
+       if (gic_rdists->has_direct_lpi &&
+           (is_v4_1(its_dev->its) || !irqd_is_forwarded_to_vcpu(d)))
                direct_lpi_inv(d);
        else if (!irqd_is_forwarded_to_vcpu(d))
                its_send_inv(its_dev, its_get_event_id(d));