diff --git a/kernel/sched/psi.c b/kernel/sched/psi.c
index eb69db4b2d6f..c41fff54324f 100644
--- a/kernel/sched/psi.c
+++ b/kernel/sched/psi.c
@@ -551,6 +551,8 @@ static u64 update_triggers(struct psi_group *group, u64 now)
 		if (now < t->last_event_time + t->win.size)
 			continue;
 
+		trace_android_vh_psi_event(t);
+
 		/* Generate an event */
 		if (cmpxchg(&t->event, 0, 1) == 0)
 			wake_up_interruptible(&t->event_wait);
@@ -559,6 +561,8 @@ static u64 update_triggers(struct psi_group *group, u64 now)
 		t->pending_event = false;
 	}
 
+	trace_android_vh_psi_group(group);
+
 	if (update_total)
 		memcpy(group->polling_total, total,
 				sizeof(group->polling_total));