memory: tegra: Register SMMU after MC driver became ready

Memory Controller driver invokes SMMU driver registration and MC's
registers mapping is shared with SMMU. This mapping goes away if MC
driver probing fails after SMMU registration.

Signed-off-by: Dmitry Osipenko <digetx@gmail.com>
Signed-off-by: Thierry Reding <treding@nvidia.com>

1 files changed, 9 insertions, 9 deletions

diff --git a/drivers/memory/tegra/mc.c b/drivers/memory/tegra/mc.c
index c81d01caf1a8..bb93cc53554e 100644
--- a/drivers/memory/tegra/mc.c
+++ b/drivers/memory/tegra/mc.c
@@ -672,15 +672,6 @@ static int tegra_mc_probe(struct platform_device *pdev)
                 return err;
         }
 
-        if (IS_ENABLED(CONFIG_TEGRA_IOMMU_SMMU)) {
-                mc->smmu = tegra_smmu_probe(&pdev->dev, mc->soc->smmu, mc);
-                if (IS_ERR(mc->smmu)) {
-                        dev_err(&pdev->dev, "failed to probe SMMU: %ld\n",
-                                PTR_ERR(mc->smmu));
-                        return PTR_ERR(mc->smmu);
-                }
-        }
-
         err = tegra_mc_reset_setup(mc);
         if (err < 0) {
                 dev_err(&pdev->dev, "failed to register reset controller: %d\n",
@@ -706,6 +697,15 @@ static int tegra_mc_probe(struct platform_device *pdev)
                 return err;
         }
 
+        if (IS_ENABLED(CONFIG_TEGRA_IOMMU_SMMU)) {
+                mc->smmu = tegra_smmu_probe(&pdev->dev, mc->soc->smmu, mc);
+                if (IS_ERR(mc->smmu)) {
+                        dev_err(&pdev->dev, "failed to probe SMMU: %ld\n",
+                                PTR_ERR(mc->smmu));
+                        return PTR_ERR(mc->smmu);
+                }
+        }
+
         return 0;
 }