1 files changed, 3 insertions, 0 deletions

diff --git a/drivers/iommu/tegra-smmu.c b/drivers/iommu/tegra-smmu.c
index 8d30653cd13a..27b1249f0773 100644
--- a/drivers/iommu/tegra-smmu.c
+++ b/drivers/iommu/tegra-smmu.c
@@ -327,6 +327,9 @@ static void tegra_smmu_domain_free(struct iommu_domain *domain)
 
         /* TODO: free page directory and page tables */
 
+        WARN_ON_ONCE(as->use_count);
+        kfree(as->count);
+        kfree(as->pts);
         kfree(as);
 }