diff --git a/drivers/irqchip/irq-gic-v3.c b/drivers/irqchip/irq-gic-v3.c
index a17b6cf9b682..a41b46bfd85b 100644
--- a/drivers/irqchip/irq-gic-v3.c
+++ b/drivers/irqchip/irq-gic-v3.c
@@ -1211,7 +1211,7 @@ static int gic_set_affinity(struct irq_data *d, const struct cpumask *mask_val,
 	reg = gic_dist_base(d) + offset + (index * 8);
 	val = gic_mpidr_to_affinity(cpu_logical_map(cpu));
 
-	trace_android_vh_gic_v3_set_affinity(d, mask_val, &val);
+	trace_android_vh_gic_v3_set_affinity(d, mask_val, &val, force, gic_dist_base(d));
 	gic_write_irouter(val, reg);
 
 	/*
diff --git a/include/trace/hooks/gic_v3.h b/include/trace/hooks/gic_v3.h
index cd2eb8df37d4..28d4c04b98ed 100644
--- a/include/trace/hooks/gic_v3.h
+++ b/include/trace/hooks/gic_v3.h
@@ -17,8 +17,8 @@ DECLARE_HOOK(android_vh_gic_v3_affinity_init,
 	TP_ARGS(irq, offset, affinity));
 DECLARE_HOOK(android_vh_gic_v3_set_affinity,
 	TP_PROTO(struct irq_data *d, const struct cpumask *mask_val,
-		 u64 *affinity),
-	TP_ARGS(d, mask_val, affinity));
+		 u64 *affinity, bool force, void __iomem *base),
+	TP_ARGS(d, mask_val, affinity, force, base));
 
 /* macro versions of hooks are no longer required */