sched_ext: selftests/dsp_local_on: Fix selftest on UP systems

commit 3c7d51b0d29954c40ea3a097e0ec7884b4344331 upstream.

In UP systems p->migration_disabled is not available. Fix this by using
the portable helper is_migration_disabled(p).

Fixes: e9fe182772dc ("sched_ext: selftests/dsp_local_on: Fix sporadic failures")
Signed-off-by: Andrea Righi <arighi@nvidia.com>
Signed-off-by: Tejun Heo <tj@kernel.org>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

diff --git a/tools/testing/selftests/sched_ext/dsp_local_on.bpf.c b/tools/testing/selftests/sched_ext/dsp_local_on.bpf.c
index eea06decb6f595afbcc652c00920efc2ac9ef955..6dcf7e6104afb09d1402a7a4322975c5a3d1cd85 100644 (file)
--- a/tools/testing/selftests/sched_ext/dsp_local_on.bpf.c
+++ b/tools/testing/selftests/sched_ext/dsp_local_on.bpf.c
@@ -43,7 +43,7 @@ void BPF_STRUCT_OPS(dsp_local_on_dispatch, s32 cpu, struct task_struct *prev)
        if (!p)
                return;
 
-       if (p->nr_cpus_allowed == nr_cpus && !p->migration_disabled)
+       if (p->nr_cpus_allowed == nr_cpus && !is_migration_disabled(p))
                target = bpf_get_prandom_u32() % nr_cpus;
        else
                target = scx_bpf_task_cpu(p);