]> Gentwo Git Trees - linux/.git/commit
sched_ext: Fix cgroup exit ordering by moving sched_ext_free() to finish_task_switch()
authorTejun Heo <tj@kernel.org>
Mon, 3 Nov 2025 20:25:13 +0000 (10:25 -1000)
committerTejun Heo <tj@kernel.org>
Mon, 3 Nov 2025 21:57:30 +0000 (11:57 -1000)
commit7900aa699c34401cf5d0c701d9ef72880ddc1a83
tree8e12a9d55ad145e2bd55d7e47a29c47aa7fec030
parent587eb08a5fef911856c16c8bb444b2ab7f4f207f
sched_ext: Fix cgroup exit ordering by moving sched_ext_free() to finish_task_switch()

sched_ext_free() was called from __put_task_struct() when the last reference
to the task is dropped, which could be long after the task has finished
running. This causes cgroup-related problems:

- ops.init_task() can be called on a cgroup which didn't get ops.cgroup_init()'d
  during scheduler load, because the cgroup might be destroyed/unlinked
  while the zombie or dead task is still lingering on the scx_tasks list.

- ops.cgroup_exit() could be called before ops.exit_task() is called on all
  member tasks, leading to incorrect exit ordering.

Fix by moving it to finish_task_switch() to be called right after the final
context switch away from the dying task, matching when sched_class->task_dead()
is called. Rename it to sched_ext_dead() to match the new calling context.

By calling sched_ext_dead() before cgroup_task_dead(), we ensure that:

- Tasks visible on scx_tasks list have valid cgroups during scheduler load,
  as cgroup_mutex prevents cgroup destruction while the task is still linked.

- All member tasks have ops.exit_task() called and are removed from scx_tasks
  before the cgroup can be destroyed and trigger ops.cgroup_exit().

This fix is made possible by the cgroup_task_dead() split in the previous patch.

This also makes more sense resource-wise as there's no point in keeping
scheduler side resources around for dead tasks.

Reported-by: Dan Schatzberg <dschatzberg@meta.com>
Cc: Peter Zijlstra <peterz@infradead.org>
Reviewed-by: Andrea Righi <arighi@nvidia.com>
Signed-off-by: Tejun Heo <tj@kernel.org>
include/linux/sched/ext.h
kernel/fork.c
kernel/sched/core.c
kernel/sched/ext.c