diff --git a/arch/arm64/mm/dma-mapping.c b/arch/arm64/mm/dma-mapping.c
index a992fb88cbed..ba8f22dbd123 100644
--- a/arch/arm64/mm/dma-mapping.c
+++ b/arch/arm64/mm/dma-mapping.c
@@ -52,7 +52,7 @@ void arch_setup_dma_ops(struct device *dev, u64 dma_base, u64 size,
 	dev->dma_coherent = coherent;
 	if (iommu) {
 		iommu_setup_dma_ops(dev, dma_base, dma_base + size - 1);
-		trace_android_vh_iommu_setup_dma_ops(dev, dma_base, size);
+		trace_android_vh_iommu_setup_dma_ops(dev, dma_base, dma_base + size - 1);
 	}
 
 #ifdef CONFIG_XEN
diff --git a/include/trace/hooks/iommu.h b/include/trace/hooks/iommu.h
index 3c5829e6f6ef..ede5271dfe12 100644
--- a/include/trace/hooks/iommu.h
+++ b/include/trace/hooks/iommu.h
@@ -15,12 +15,12 @@
 #if defined(CONFIG_TRACEPOINTS) && defined(CONFIG_ANDROID_VENDOR_HOOKS)
 
 DECLARE_HOOK(android_vh_iommu_setup_dma_ops,
-	TP_PROTO(struct device *dev, u64 dma_base, u64 size),
-	TP_ARGS(dev, dma_base, size));
+	TP_PROTO(struct device *dev, u64 dma_base, u64 dma_limit),
+	TP_ARGS(dev, dma_base, dma_limit));
 
 #else
 
-#define trace_android_vh_iommu_setup_dma_ops(dev, dma_base, size)
+#define trace_android_vh_iommu_setup_dma_ops(dev, dma_base, dma_limit)
 
 #endif