diff --git a/drivers/iommu/dma-iommu.c b/drivers/iommu/dma-iommu.c
index eabfeb59b5e7..a6e6068506c0 100644
--- a/drivers/iommu/dma-iommu.c
+++ b/drivers/iommu/dma-iommu.c
@@ -1759,6 +1759,8 @@ void iommu_setup_dma_ops(struct device *dev)
 	if (dev->dma_iommu && iommu_dma_init_domain(domain, dev))
 		goto out_err;
 
+	trace_android_rvh_iommu_setup_dma_ops(dev);
+
 	return;
 out_err:
 	pr_warn("Failed to set up IOMMU for device %s; retaining platform DMA ops\n",
diff --git a/include/trace/hooks/iommu.h b/include/trace/hooks/iommu.h
index 6357a6bc8f96..d965a23c1c92 100644
--- a/include/trace/hooks/iommu.h
+++ b/include/trace/hooks/iommu.h
@@ -10,8 +10,8 @@
 #include <trace/hooks/vendor_hooks.h>
 
 DECLARE_RESTRICTED_HOOK(android_rvh_iommu_setup_dma_ops,
-	TP_PROTO(struct device *dev, u64 dma_base, u64 dma_limit),
-	TP_ARGS(dev, dma_base, dma_limit), 1);
+	TP_PROTO(struct device *dev),
+	TP_ARGS(dev), 1);
 
 struct iova_domain;