* of_msi_map_get_device_domain - Use msi-map to find the relevant MSI domain
  * @dev: device for which the mapping is to be done.
  * @rid: Requester ID for the device.
+ * @bus_token: Bus token
  *
  * Walk up the device hierarchy looking for devices with a "msi-map"
  * property.
  *
  * Returns: the MSI domain for this device (or NULL on failure)
  */
-struct irq_domain *of_msi_map_get_device_domain(struct device *dev, u32 rid)
+struct irq_domain *of_msi_map_get_device_domain(struct device *dev, u32 id,
+                                               u32 bus_token)
 {
        struct device_node *np = NULL;
 
-       __of_msi_map_rid(dev, &np, rid);
-       return irq_find_matching_host(np, DOMAIN_BUS_PCI_MSI);
+       __of_msi_map_rid(dev, &np, id);
+       return irq_find_matching_host(np, bus_token);
 }
 
 /**
 
        u32 rid = pci_dev_id(pdev);
 
        pci_for_each_dma_alias(pdev, get_msi_id_cb, &rid);
-       dom = of_msi_map_get_device_domain(&pdev->dev, rid);
+       dom = of_msi_map_get_device_domain(&pdev->dev, rid, DOMAIN_BUS_PCI_MSI);
        if (!dom)
                dom = iort_get_device_domain(&pdev->dev, rid,
                                             DOMAIN_BUS_PCI_MSI);
 
                                            struct device_node *np,
                                            enum irq_domain_bus_token token);
 extern struct irq_domain *of_msi_map_get_device_domain(struct device *dev,
-                                                      u32 rid);
+                                                       u32 id,
+                                                       u32 bus_token);
 extern void of_msi_configure(struct device *dev, struct device_node *np);
 u32 of_msi_map_rid(struct device *dev, struct device_node *msi_np, u32 rid_in);
 #else
        return NULL;
 }
 static inline struct irq_domain *of_msi_map_get_device_domain(struct device *dev,
-                                                             u32 rid)
+                                               u32 id, u32 bus_token)
 {
        return NULL;
 }