Re: [PATCH 2/6] sched_ext: idle: Explicitly pass allowed cpumask to scx_select_cpu_dfl()

[Andrea Righi <arighi@nvidia.com>]

On Fri, Mar 21, 2025 at 07:15:37PM +0900, changwoo wrote:
> Hi Andrea,
> 
> On 3/20/25 16:36, Andrea Righi wrote:
> > Modify scx_select_cpu_dfl() to take the allowed cpumask as an explicit
> > argument, instead of implicitly using @p->cpus_ptr.
> > 
> > This prepares for future changes where arbitrary cpumasks may be passed
> > to the built-in idle CPU selection policy.
> > 
> > This is a pure refactoring with no functional changes.
> > 
> > Signed-off-by: Andrea Righi <arighi@nvidia.com>
> > ---
> >   kernel/sched/ext.c      |  2 +-
> >   kernel/sched/ext_idle.c | 45 ++++++++++++++++++++++++++---------------
> >   kernel/sched/ext_idle.h |  3 ++-
> >   3 files changed, 32 insertions(+), 18 deletions(-)
> > 
> > diff --git a/kernel/sched/ext.c b/kernel/sched/ext.c
> > index 06561d6717c9a..f42352e8d889e 100644
> > --- a/kernel/sched/ext.c
> > +++ b/kernel/sched/ext.c
> > @@ -3395,7 +3395,7 @@ static int select_task_rq_scx(struct task_struct *p, int prev_cpu, int wake_flag
> >   	} else {
> >   		s32 cpu;
> >   
> > -		cpu = scx_select_cpu_dfl(p, prev_cpu, wake_flags, 0);
> > +		cpu = scx_select_cpu_dfl(p, prev_cpu, wake_flags, p->cpus_ptr, 0);
> >   		if (cpu >= 0) {
> >   			p->scx.slice = SCX_SLICE_DFL;
> >   			p->scx.ddsp_dsq_id = SCX_DSQ_LOCAL;
> > diff --git a/kernel/sched/ext_idle.c b/kernel/sched/ext_idle.c
> > index e1e020c27c07c..a90d85bce1ccb 100644
> > --- a/kernel/sched/ext_idle.c
> > +++ b/kernel/sched/ext_idle.c
> > @@ -397,11 +397,19 @@ void scx_idle_update_selcpu_topology(struct sched_ext_ops *ops)
> >   		static_branch_disable_cpuslocked(&scx_selcpu_topo_numa);
> >   }
> >   
> > +static inline bool task_allowed_all_cpus(const struct task_struct *p)
> > +{
> > +	return p->nr_cpus_allowed >= num_possible_cpus();
> > +}
> 
> This function will be renamed to task_affinity_all() in patch #3.
> Can we use the same name from the beginning?
> That will make the commits easier to read.

Right, I'll clean up also this patch in the next version, thanks!

-Andrea