diff --git a/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.c b/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.c
index e02bf2578a64969f7f3eac283a8e62f328d07d76..d3575e05b1bea8673b2ab21cd1c80378d705f7b3 100644
--- a/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.c
+++ b/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.c
@@ -3832,6 +3832,7 @@ arm_smmu_cache_invalidate(struct iommu_domain *domain, struct device *dev,
 		    !(granule_size & smmu_domain->domain.pgsize_bitmap)) {
 			tg = __ffs(smmu_domain->domain.pgsize_bitmap);
 			granule_size = 1 << tg;
+			size = size >> tg;
 		}
 
 		arm_smmu_tlb_inv_range_domain(info->addr, size,