diff --git a/include/trace/hooks/sched.h b/include/trace/hooks/sched.h
index d0d0265cdd76..b5bb3186ac5f 100644
--- a/include/trace/hooks/sched.h
+++ b/include/trace/hooks/sched.h
@@ -65,8 +65,8 @@ DECLARE_RESTRICTED_HOOK(android_rvh_rto_next_cpu,
 	TP_ARGS(rto_cpu, rto_mask, cpu), 1);
 
 DECLARE_RESTRICTED_HOOK(android_rvh_is_cpu_allowed,
-	TP_PROTO(int cpu, bool *allowed),
-	TP_ARGS(cpu, allowed), 1);
+	TP_PROTO(struct task_struct *p, int cpu, bool *allowed),
+	TP_ARGS(p, cpu, allowed), 1);
 
 DECLARE_RESTRICTED_HOOK(android_rvh_get_nohz_timer_target,
 	TP_PROTO(int *cpu, bool *done),
diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index 630553aaa4b5..0e98ac84d8d1 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -2230,7 +2230,7 @@ static inline bool is_cpu_allowed(struct task_struct *p, int cpu)
 		return cpu_online(cpu);
 
 	/* check for all cases */
-	trace_android_rvh_is_cpu_allowed(cpu, &allowed);
+	trace_android_rvh_is_cpu_allowed(p, cpu, &allowed);
 
 	/* Non kernel threads are not allowed during either online or offline. */
 	if (!(p->flags & PF_KTHREAD))