1337d1b35SAndrea Righi /* SPDX-License-Identifier: GPL-2.0 */ 2337d1b35SAndrea Righi /* 3337d1b35SAndrea Righi * BPF extensible scheduler class: Documentation/scheduler/sched-ext.rst 4337d1b35SAndrea Righi * 5337d1b35SAndrea Righi * Copyright (c) 2022 Meta Platforms, Inc. and affiliates. 6337d1b35SAndrea Righi * Copyright (c) 2022 Tejun Heo <tj@kernel.org> 7337d1b35SAndrea Righi * Copyright (c) 2022 David Vernet <dvernet@meta.com> 8337d1b35SAndrea Righi * Copyright (c) 2024 Andrea Righi <arighi@nvidia.com> 9337d1b35SAndrea Righi */ 10337d1b35SAndrea Righi #ifndef _KERNEL_SCHED_EXT_IDLE_H 11337d1b35SAndrea Righi #define _KERNEL_SCHED_EXT_IDLE_H 12337d1b35SAndrea Righi 13d73249f8SAndrea Righi struct sched_ext_ops; 14337d1b35SAndrea Righi 15337d1b35SAndrea Righi #ifdef CONFIG_SMP 160aaaf89dSAndrea Righi void scx_idle_update_selcpu_topology(struct sched_ext_ops *ops); 17337d1b35SAndrea Righi void scx_idle_init_masks(void); 18337d1b35SAndrea Righi bool scx_idle_test_and_clear_cpu(int cpu); 1948849271SAndrea Righi s32 scx_pick_idle_cpu(const struct cpumask *cpus_allowed, int node, u64 flags); 20337d1b35SAndrea Righi #else /* !CONFIG_SMP */ 210aaaf89dSAndrea Righi static inline void scx_idle_update_selcpu_topology(struct sched_ext_ops *ops) {} 22337d1b35SAndrea Righi static inline void scx_idle_init_masks(void) {} 23337d1b35SAndrea Righi static inline bool scx_idle_test_and_clear_cpu(int cpu) { return false; } 2448849271SAndrea Righi static inline s32 scx_pick_idle_cpu(const struct cpumask *cpus_allowed, int node, u64 flags) 25337d1b35SAndrea Righi { 26337d1b35SAndrea Righi return -EBUSY; 27337d1b35SAndrea Righi } 28337d1b35SAndrea Righi #endif /* CONFIG_SMP */ 29337d1b35SAndrea Righi 30*e4855fc9SAndrea Righi s32 scx_select_cpu_dfl(struct task_struct *p, s32 prev_cpu, u64 wake_flags, u64 flags); 31d73249f8SAndrea Righi void scx_idle_enable(struct sched_ext_ops *ops); 32d73249f8SAndrea Righi void scx_idle_disable(void); 33d73249f8SAndrea Righi int scx_idle_init(void); 34337d1b35SAndrea Righi 35337d1b35SAndrea Righi #endif /* _KERNEL_SCHED_EXT_IDLE_H */ 36