1 files changed, 8 insertions, 5 deletions

diff --git a/drivers/iommu/tegra-smmu.c b/drivers/iommu/tegra-smmu.c
index a9bf4c30cd84..37669cc483b1 100644
--- a/drivers/iommu/tegra-smmu.c
+++ b/drivers/iommu/tegra-smmu.c
@@ -814,11 +814,14 @@ static int smmu_iommu_domain_init(struct iommu_domain *domain)
         /* Look for a free AS with lock held */
         for  (i = 0; i < smmu->num_as; i++) {
                 as = &smmu->as[i];
-                if (!as->pdir_page) {
-                        err = alloc_pdir(as);
-                        if (!err)
-                                goto found;
-                }
+
+                if (as->pdir_page)
+                        continue;
+
+                err = alloc_pdir(as);
+                if (!err)
+                        goto found;
+
                 if (err != -EAGAIN)
                         break;
         }