From 1e0516e27bb23283030d818092b055a66b0ff11e Mon Sep 17 00:00:00 2001
From: Ze Zuo <zuoze1@huawei.com>
Date: Sun, 18 May 2025 16:34:41 +0800
Subject: [PATCH] sched: Enable per-process mem_sampling from sched switch path

hulk inclusion
category:feature
bugzilla:https://gitee.com/openeuler/kernel/issues/IC8KS8
CVE: NA

--------------------------------

Program mem_sampling for access profiling for threads from the
task sched switch path. mem_sampling is programmed with a period
that corresponds to the incoming thread. Kernel threads are
excluded from this.

Signed-off-by: Ze Zuo <zuoze1@huawei.com>
Signed-off-by: Tong Tiangen <tongtiangen@huawei.com>
Signed-off-by: Shuang Yan <yanshuang7@huawei.com>
---
 kernel/sched/core.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index 7dc4ceebd5ec..7c40690ad56f 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -64,6 +64,7 @@
 #include <linux/vtime.h>
 #include <linux/wait_api.h>
 #include <linux/workqueue_api.h>
+#include <linux/mem_sampling.h>
 
 #ifdef CONFIG_PREEMPT_DYNAMIC
 # ifdef CONFIG_GENERIC_ENTRY
@@ -5307,6 +5308,7 @@ static struct rq *finish_task_switch(struct task_struct *prev)
 	prev_state = READ_ONCE(prev->__state);
 	vtime_task_switch(prev);
 	perf_event_task_sched_in(prev, current);
+	mem_sampling_sched_in(prev, current);
 	finish_task(prev);
 	tick_nohz_task_switch();
 	finish_lock_switch(rq);
-- 
Gitee