Commit 5f52bbf2 authored by Andrea Righi's avatar Andrea Righi Committed by Tejun Heo
Browse files

tools/sched_ext: Add helper to check task migration state 



Introduce a new helper for BPF schedulers to determine whether a task
can migrate or not (supporting both SMP and UP systems).

Fixes: e9fe1827 ("sched_ext: selftests/dsp_local_on: Fix sporadic failures")
Signed-off-by: default avatarAndrea Righi <arighi@nvidia.com>
Signed-off-by: default avatarTejun Heo <tj@kernel.org>
parent d6f3e7d5

tools/sched_ext/include/scx/common.bpf.h

+11 −0
Original line number Diff line number Diff line
@@ -404,6 +404,17 @@ static __always_inline const struct cpumask *cast_mask(struct bpf_cpumask *mask) 
	return (const struct cpumask *)mask;

}



/*

 * Return true if task @p cannot migrate to a different CPU, false

 * otherwise.

 */

static inline bool is_migration_disabled(const struct task_struct *p)

{

	if (bpf_core_field_exists(p->migration_disabled))

		return p->migration_disabled;

	return false;

}



/* rcu */

void bpf_rcu_read_lock(void) __ksym;

void bpf_rcu_read_unlock(void) __ksym;