*/
 struct amd_iommu **amd_iommu_rlookup_table;
 
-/*
- * This table is used to find the irq remapping table for a given device id
- * quickly.
- */
-struct irq_remap_table **irq_lookup_table;
-
 /*
  * AMD IOMMU allows up to 2^16 different protection domains. This is a bitmap
  * to know which ones are already in use.
 
 static void __init free_iommu_resources(void)
 {
-       kmemleak_free(irq_lookup_table);
-       free_pages((unsigned long)irq_lookup_table,
-                  get_order(rlookup_table_size));
-       irq_lookup_table = NULL;
-
        kmem_cache_destroy(amd_iommu_irq_cache);
        amd_iommu_irq_cache = NULL;
 
                        if (alloc_irq_lookup_table(pci_seg))
                                goto out;
                }
-
-               irq_lookup_table = (void *)__get_free_pages(
-                               GFP_KERNEL | __GFP_ZERO,
-                               get_order(rlookup_table_size));
-               kmemleak_alloc(irq_lookup_table, rlookup_table_size,
-                              1, GFP_KERNEL);
-               if (!irq_lookup_table)
-                       goto out;
        }
 
        ret = init_memory_definitions(ivrs_base);
 
        amd_iommu_dev_table[devid].data[2] = dte;
 }
 
-static struct irq_remap_table *get_irq_table(u16 devid)
+static struct irq_remap_table *get_irq_table(struct amd_iommu *iommu, u16 devid)
 {
        struct irq_remap_table *table;
+       struct amd_iommu_pci_seg *pci_seg = iommu->pci_seg;
 
        if (WARN_ONCE(!amd_iommu_rlookup_table[devid],
                      "%s: no iommu for devid %x\n", __func__, devid))
                return NULL;
 
-       table = irq_lookup_table[devid];
-       if (WARN_ONCE(!table, "%s: no table for devid %x\n", __func__, devid))
+       table = pci_seg->irq_lookup_table[devid];
+       if (WARN_ONCE(!table, "%s: no table for devid %x:%x\n",
+                     __func__, pci_seg->id, devid))
                return NULL;
 
        return table;
 static void set_remap_table_entry(struct amd_iommu *iommu, u16 devid,
                                  struct irq_remap_table *table)
 {
-       irq_lookup_table[devid] = table;
+       struct amd_iommu_pci_seg *pci_seg = iommu->pci_seg;
+
+       pci_seg->irq_lookup_table[devid] = table;
        set_dte_irq_entry(devid, table);
        iommu_flush_dte(iommu, devid);
 }
                                       void *data)
 {
        struct irq_remap_table *table = data;
+       struct amd_iommu_pci_seg *pci_seg;
+       struct amd_iommu *iommu = rlookup_amd_iommu(&pdev->dev);
 
-       irq_lookup_table[alias] = table;
+       if (!iommu)
+               return -EINVAL;
+
+       pci_seg = iommu->pci_seg;
+       pci_seg->irq_lookup_table[alias] = table;
        set_dte_irq_entry(alias, table);
 
        iommu_flush_dte(amd_iommu_rlookup_table[alias], alias);
                goto out_unlock;
 
        pci_seg = iommu->pci_seg;
-       table = irq_lookup_table[devid];
+       table = pci_seg->irq_lookup_table[devid];
        if (table)
                goto out_unlock;
 
        alias = pci_seg->alias_table[devid];
-       table = irq_lookup_table[alias];
+       table = pci_seg->irq_lookup_table[alias];
        if (table) {
                set_remap_table_entry(iommu, devid, table);
                goto out_wait;
 
        spin_lock_irqsave(&iommu_table_lock, flags);
 
-       table = irq_lookup_table[devid];
+       table = pci_seg->irq_lookup_table[devid];
        if (table)
                goto out_unlock;
 
-       table = irq_lookup_table[alias];
+       table = pci_seg->irq_lookup_table[alias];
        if (table) {
                set_remap_table_entry(iommu, devid, table);
                goto out_wait;
        if (iommu == NULL)
                return -EINVAL;
 
-       table = get_irq_table(devid);
+       table = get_irq_table(iommu, devid);
        if (!table)
                return -ENOMEM;
 
        if (iommu == NULL)
                return -EINVAL;
 
-       table = get_irq_table(devid);
+       table = get_irq_table(iommu, devid);
        if (!table)
                return -ENOMEM;
 
        if (iommu == NULL)
                return;
 
-       table = get_irq_table(devid);
+       table = get_irq_table(iommu, devid);
        if (!table)
                return;
 
        if (!iommu)
                return -ENODEV;
 
-       table = get_irq_table(devid);
+       table = get_irq_table(iommu, devid);
        if (!table)
                return -ENODEV;