sched/fair: Simplify task_numa_find_cpu()

Use for_each_cpu_and() and drop some housekeeping code.

Signed-off-by: Yury Norov (NVIDIA) <yury.norov@gmail.com>
Signed-off-by: Peter Zijlstra (Intel) <peterz@infradead.org>
Reviewed-by: K Prateek Nayak <kprateek.nayak@amd.com>
Reviewed-by: Phil Auld <pauld@redhat.com>
Link: https://patch.msgid.link/20251207033037.399608-1-yury.norov@gmail.com

diff --git a/kernel/sched/fair.c b/kernel/sched/fair.c
index 64275d75a964d9071bfb18adfadd9a4a83063cec..842a0f20414f1ae88eccee7ac56d28a02c0cf5fc 100644 (file)
--- a/kernel/sched/fair.c
+++ b/kernel/sched/fair.c
@@ -2494,11 +2494,8 @@ static void task_numa_find_cpu(struct task_numa_env *env,
                maymove = !load_too_imbalanced(src_load, dst_load, env);
        }
 
-       for_each_cpu(cpu, cpumask_of_node(env->dst_nid)) {
-               /* Skip this CPU if the source task cannot migrate */
-               if (!cpumask_test_cpu(cpu, env->p->cpus_ptr))
-                       continue;
-
+       /* Skip CPUs if the source task cannot migrate */
+       for_each_cpu_and(cpu, cpumask_of_node(env->dst_nid), env->p->cpus_ptr) {
                env->dst_cpu = cpu;
                if (task_numa_compare(env, taskimp, groupimp, maymove))
                        break;