2 files changed, 10 insertions, 0 deletions

diff --git a/drivers/iommu/tegra-gart.c b/drivers/iommu/tegra-gart.c
index 0c0a37792218..c16e8fc8a4bd 100644
--- a/drivers/iommu/tegra-gart.c
+++ b/drivers/iommu/tegra-gart.c
@@ -165,6 +165,11 @@ static int gart_iommu_attach_dev(struct iommu_domain *domain,
                 return -EINVAL;
         domain->priv = gart;
 
+        domain->geometry.aperture_start = gart->iovmm_base;
+        domain->geometry.aperture_end   = gart->iovmm_base +
+                                        gart->page_count * GART_PAGE_SIZE - 1;
+        domain->geometry.force_aperture = true;
+
         client = devm_kzalloc(gart->dev, sizeof(*c), GFP_KERNEL);
         if (!client)
                 return -ENOMEM;
diff --git a/drivers/iommu/tegra-smmu.c b/drivers/iommu/tegra-smmu.c
index ecd679043d77..96e73d56451a 100644
--- a/drivers/iommu/tegra-smmu.c
+++ b/drivers/iommu/tegra-smmu.c
@@ -807,6 +807,11 @@ found:
         spin_unlock_irqrestore(&as->lock, flags);
         domain->priv = as;
 
+        domain->geometry.aperture_start = smmu->iovmm_base;
+        domain->geometry.aperture_end   = smmu->iovmm_base +
+                smmu->page_count * SMMU_PAGE_SIZE - 1;
+        domain->geometry.force_aperture = true;
+
         dev_dbg(smmu->dev, "smmu_as@%p\n", as);
         return 0;