Re: [PATCH v33 4/7] sched/proxy: Set TASK_RUNNING when shortcuting return migration

[Andrea Righi <arighi@nvidia.com>]

Hi John,

On Fri, Oct 23, 2026 at 10:04:15PM +0000, John Stultz wrote:
> K Prateek's prior patch optimizing find_proxy_tasks() logic to
> shortcut and run previously blocked tasks on the current cpu
> made me realize while we are clearing the blocked_on and
> is_blocked, we are not setting the task state as TASK_RUNNING
> before returning the task to run.
> 
> So set the tasks to TASK_RUNNING prior to returning them to run.
> 
> Fixes: b049b81bdff6 ("sched: Handle blocked-waiter migration (and return migration)")
> Signed-off-by: John Stultz <jstultz@google.com>
> ---
> Cc: Peter Zijlstra <peterz@infradead.org>
> Cc: Ingo Molnar <mingo@redhat.com>
> Cc: K Prateek Nayak <kprateek.nayak@amd.com>
> Cc: Thomas Gleixner <tglx@linutronix.de>
> Cc: Juri Lelli <juri.lelli@redhat.com>
> Cc: Vincent Guittot <vincent.guittot@linaro.org>
> Cc: Valentin Schneider <vschneid@redhat.com>
> Cc: Suleiman Souhlal <suleiman@google.com>
> Cc: Andrea Righi <arighi@nvidia.com>
> Cc: Metin Kaya <Metin.Kaya@arm.com>
> Cc: Dietmar Eggemann <dietmar.eggemann@arm.com>
> Cc: Steven Rostedt <rostedt@goodmis.org>
> Cc: Zimuzo Ezeozue <zezeozue@google.com>
> Cc: Waiman Long <longman@redhat.com>
> Cc: Boqun Feng <boqun.feng@gmail.com>
> Cc: "Paul E. McKenney" <paulmck@kernel.org>
> Cc: Joel Fernandes <joelagnelf@nvidia.com>
> Cc: Qais Yousef <qyousef@layalina.io>
> Cc: Xuewen Yan <xuewen.yan94@gmail.com>
> Cc: kuyo chang <kuyo.chang@mediatek.com>
> Cc: hupu <hupu.gm@gmail.com>
> Cc: Zhan Xusheng <zhanxusheng@xiaomi.com>
> Cc: zhidao su (Xiaomi) <soolaugust@gmail.com>
> Cc: Hui Su <sh_def@163.com>
> Cc: Shubhang Kaushik (Ampere) <sh@gentwo.org>
> Cc: kernel-team@android.com
> ---
>  kernel/sched/core.c | 2 ++
>  1 file changed, 2 insertions(+)
> 
> diff --git a/kernel/sched/core.c b/kernel/sched/core.c
> index 2cb42de255f7a..43ba76b5a3464 100644
> --- a/kernel/sched/core.c
> +++ b/kernel/sched/core.c
> @@ -7028,6 +7028,7 @@ find_proxy_task(struct rq *rq, struct task_struct *donor, struct rq_flags *rf)
>  			clear_task_blocked_on(p, mutex);
>  			if (task_current(rq, p) || p->wake_cpu == task_cpu(p)) {
>  				p->is_blocked = 0;
> +				WRITE_ONCE(p->__state, TASK_RUNNING);
>  				return p;
>  			}
>  			goto activate;

This is basically an inlined ttwu_do_wakeup() without
trace_sched_wakeup().

> @@ -7065,6 +7066,7 @@ find_proxy_task(struct rq *rq, struct task_struct *donor, struct rq_flags *rf)
>  			__clear_task_blocked_on(p, NULL);
>  			if (task_current(rq, p) || p->wake_cpu == task_cpu(p)) {
>  				p->is_blocked = 0;
> +				WRITE_ONCE(p->__state, TASK_RUNNING);
>  				return p;
>  			}
>  			goto activate;

Ditto.

Can we just call ttwu_do_wakeup(p)? It'd also avoid having two copies of
the same state transition that can diverge.

Thanks,
-Andrea