diff --git a/drivers/android/vendor_hooks.c b/drivers/android/vendor_hooks.c index f1c536a29ef2..96f63b6acbf0 100644 --- a/drivers/android/vendor_hooks.c +++ b/drivers/android/vendor_hooks.c @@ -197,3 +197,4 @@ EXPORT_TRACEPOINT_SYMBOL_GPL(android_rvh_rto_next_cpu); EXPORT_TRACEPOINT_SYMBOL_GPL(android_vh_em_dev_register_pd); EXPORT_TRACEPOINT_SYMBOL_GPL(android_vh_logbuf); EXPORT_TRACEPOINT_SYMBOL_GPL(android_vh_logbuf_pr_cont); +EXPORT_TRACEPOINT_SYMBOL_GPL(android_rvh_post_init_entity_util_avg); diff --git a/include/trace/hooks/sched.h b/include/trace/hooks/sched.h index b5f2633e08bd..06f0e5e58279 100644 --- a/include/trace/hooks/sched.h +++ b/include/trace/hooks/sched.h @@ -323,6 +323,10 @@ DECLARE_HOOK(android_vh_em_dev_register_pd, TP_PROTO(bool *cond), TP_ARGS(cond)); +DECLARE_RESTRICTED_HOOK(android_rvh_post_init_entity_util_avg, + TP_PROTO(struct sched_entity *se), + TP_ARGS(se), 1); + #endif /* _TRACE_HOOK_SCHED_H */ /* This part must be outside protection */ #include diff --git a/kernel/sched/fair.c b/kernel/sched/fair.c index ebd00235d5b5..4faad3f099b9 100644 --- a/kernel/sched/fair.c +++ b/kernel/sched/fair.c @@ -809,6 +809,8 @@ void post_init_entity_util_avg(struct task_struct *p) return; } + /* Hook before this se's util is attached to cfs_rq's util */ + trace_android_rvh_post_init_entity_util_avg(se); attach_entity_cfs_rq(se); }