iommu/tegra: gart: Set aperture at domain initialization time

The aperture of the domain should always be available, otherwise drivers
need to attach first before they can use the aperture geometry.

Cc: Hiroshi Doyu <hdoyu@nvidia.com>
Signed-off-by: Thierry Reding <treding@nvidia.com>
Signed-off-by: Joerg Roedel <jroedel@suse.de>

1 files changed, 14 insertions, 11 deletions

diff --git a/drivers/iommu/tegra-gart.c b/drivers/iommu/tegra-gart.c
index c48da057dbb1..5f3b68c26b83 100644
--- a/drivers/iommu/tegra-gart.c
+++ b/drivers/iommu/tegra-gart.c
@@ -156,20 +156,10 @@ static inline bool gart_iova_range_valid(struct gart_device *gart,
 static int gart_iommu_attach_dev(struct iommu_domain *domain,
                                  struct device *dev)
 {
-        struct gart_device *gart;
+        struct gart_device *gart = domain->priv;
         struct gart_client *client, *c;
         int err = 0;
 
-        gart = gart_handle;
-        if (!gart)
-                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;
@@ -218,6 +208,19 @@ out:
 
 static int gart_iommu_domain_init(struct iommu_domain *domain)
 {
+        struct gart_device *gart;
+
+        gart = gart_handle;
+        if (!gart)
+                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;
+
         return 0;
 }