1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3 * BPF extensible scheduler class: Documentation/scheduler/sched-ext.rst
4 *
5 * Copyright (c) 2022 Meta Platforms, Inc. and affiliates.
6 * Copyright (c) 2022 Tejun Heo <tj@kernel.org>
7 * Copyright (c) 2022 David Vernet <dvernet@meta.com>
8 * Copyright (c) 2024 Andrea Righi <arighi@nvidia.com>
9 */
10 #ifndef _KERNEL_SCHED_EXT_IDLE_H
11 #define _KERNEL_SCHED_EXT_IDLE_H
12
13 struct sched_ext_ops;
14
15 #ifdef CONFIG_SMP
16 void scx_idle_update_selcpu_topology(struct sched_ext_ops *ops);
17 void scx_idle_init_masks(void);
18 bool scx_idle_test_and_clear_cpu(int cpu);
19 s32 scx_pick_idle_cpu(const struct cpumask *cpus_allowed, int node, u64 flags);
20 #else /* !CONFIG_SMP */
scx_idle_update_selcpu_topology(struct sched_ext_ops * ops)21 static inline void scx_idle_update_selcpu_topology(struct sched_ext_ops *ops) {}
scx_idle_init_masks(void)22 static inline void scx_idle_init_masks(void) {}
scx_idle_test_and_clear_cpu(int cpu)23 static inline bool scx_idle_test_and_clear_cpu(int cpu) { return false; }
scx_pick_idle_cpu(const struct cpumask * cpus_allowed,int node,u64 flags)24 static inline s32 scx_pick_idle_cpu(const struct cpumask *cpus_allowed, int node, u64 flags)
25 {
26 return -EBUSY;
27 }
28 #endif /* CONFIG_SMP */
29
30 s32 scx_select_cpu_dfl(struct task_struct *p, s32 prev_cpu, u64 wake_flags, u64 flags);
31 void scx_idle_enable(struct sched_ext_ops *ops);
32 void scx_idle_disable(void);
33 int scx_idle_init(void);
34
35 #endif /* _KERNEL_SCHED_EXT_IDLE_H */
36