1 files changed, 6 insertions, 0 deletions

diff --git a/drivers/base/iommu.c b/drivers/base/iommu.c
index cf7cbec116ed..55d37e4609eb 100644
--- a/drivers/base/iommu.c
+++ b/drivers/base/iommu.c
@@ -119,6 +119,9 @@ int iommu_map(struct iommu_domain *domain, unsigned long iova,
 
         BUG_ON((iova | paddr) & invalid_mask);
 
+        if (iommu_ops->map)
+                return iommu_ops->map(domain, iova, paddr, gfp_order, prot);
+
         return iommu_ops->map_range(domain, iova, paddr, size, prot);
 }
 EXPORT_SYMBOL_GPL(iommu_map);
@@ -133,6 +136,9 @@ int iommu_unmap(struct iommu_domain *domain, unsigned long iova, int gfp_order)
 
         BUG_ON(iova & invalid_mask);
 
+        if (iommu_ops->unmap)
+                return iommu_ops->unmap(domain, iova, gfp_order);
+
         iommu_ops->unmap_range(domain, iova, size);
 
         return gfp_order;