1 files changed, 3 insertions, 7 deletions

diff --git a/drivers/iommu/iommu.c b/drivers/iommu/iommu.c
index b9df1411c894..829b5e981217 100644
--- a/drivers/iommu/iommu.c
+++ b/drivers/iommu/iommu.c
@@ -660,8 +660,8 @@ static struct iommu_group *get_pci_function_alias_group(struct pci_dev *pdev,
 }
 
 /*
- * Look for aliases to or from the given device for exisiting groups.  The
- * dma_alias_devfn only supports aliases on the same bus, therefore the search
+ * Look for aliases to or from the given device for existing groups. DMA
+ * aliases are only supported on the same bus, therefore the search
  * space is quite small (especially since we're really only looking at pcie
  * device, and therefore only expect multiple slots on the root complex or
  * downstream switch ports).  It's conceivable though that a pair of
@@ -686,11 +686,7 @@ static struct iommu_group *get_pci_alias_group(struct pci_dev *pdev,
                         continue;
 
                 /* We alias them or they alias us */
-                if (((pdev->dev_flags & PCI_DEV_FLAGS_DMA_ALIAS_DEVFN) &&
-                     pdev->dma_alias_devfn == tmp->devfn) ||
-                    ((tmp->dev_flags & PCI_DEV_FLAGS_DMA_ALIAS_DEVFN) &&
-                     tmp->dma_alias_devfn == pdev->devfn)) {
-
+                if (pci_devs_are_dma_aliases(pdev, tmp)) {
                         group = get_pci_alias_group(tmp, devfns);
                         if (group) {
                                 pci_dev_put(tmp);