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 */ 9 #ifdef CONFIG_SCHED_CLASS_EXT 10 11 void scx_tick(struct rq *rq); 12 void init_scx_entity(struct sched_ext_entity *scx); 13 void scx_pre_fork(struct task_struct *p); 14 int scx_fork(struct task_struct *p); 15 void scx_post_fork(struct task_struct *p); 16 void scx_cancel_fork(struct task_struct *p); 17 bool scx_can_stop_tick(struct rq *rq); 18 void scx_rq_activate(struct rq *rq); 19 void scx_rq_deactivate(struct rq *rq); 20 int scx_check_setscheduler(struct task_struct *p, int policy); 21 bool task_should_scx(struct task_struct *p); 22 void init_sched_ext_class(void); 23 24 static inline u32 scx_cpuperf_target(s32 cpu) 25 { 26 if (scx_enabled()) 27 return cpu_rq(cpu)->scx.cpuperf_target; 28 else 29 return 0; 30 } 31 32 static inline bool task_on_scx(const struct task_struct *p) 33 { 34 return scx_enabled() && p->sched_class == &ext_sched_class; 35 } 36 37 #ifdef CONFIG_SCHED_CORE 38 bool scx_prio_less(const struct task_struct *a, const struct task_struct *b, 39 bool in_fi); 40 #endif 41 42 #else /* CONFIG_SCHED_CLASS_EXT */ 43 44 static inline void scx_tick(struct rq *rq) {} 45 static inline void scx_pre_fork(struct task_struct *p) {} 46 static inline int scx_fork(struct task_struct *p) { return 0; } 47 static inline void scx_post_fork(struct task_struct *p) {} 48 static inline void scx_cancel_fork(struct task_struct *p) {} 49 static inline u32 scx_cpuperf_target(s32 cpu) { return 0; } 50 static inline bool scx_can_stop_tick(struct rq *rq) { return true; } 51 static inline void scx_rq_activate(struct rq *rq) {} 52 static inline void scx_rq_deactivate(struct rq *rq) {} 53 static inline int scx_check_setscheduler(struct task_struct *p, int policy) { return 0; } 54 static inline bool task_on_scx(const struct task_struct *p) { return false; } 55 static inline void init_sched_ext_class(void) {} 56 57 #endif /* CONFIG_SCHED_CLASS_EXT */ 58 59 #if defined(CONFIG_SCHED_CLASS_EXT) && defined(CONFIG_SMP) 60 void __scx_update_idle(struct rq *rq, bool idle); 61 62 static inline void scx_update_idle(struct rq *rq, bool idle) 63 { 64 if (scx_enabled()) 65 __scx_update_idle(rq, idle); 66 } 67 #else 68 static inline void scx_update_idle(struct rq *rq, bool idle) {} 69 #endif 70 71 #ifdef CONFIG_CGROUP_SCHED 72 #ifdef CONFIG_EXT_GROUP_SCHED 73 int scx_tg_online(struct task_group *tg); 74 void scx_tg_offline(struct task_group *tg); 75 int scx_cgroup_can_attach(struct cgroup_taskset *tset); 76 void scx_move_task(struct task_struct *p); 77 void scx_cgroup_finish_attach(void); 78 void scx_cgroup_cancel_attach(struct cgroup_taskset *tset); 79 void scx_group_set_weight(struct task_group *tg, unsigned long cgrp_weight); 80 void scx_group_set_idle(struct task_group *tg, bool idle); 81 #else /* CONFIG_EXT_GROUP_SCHED */ 82 static inline int scx_tg_online(struct task_group *tg) { return 0; } 83 static inline void scx_tg_offline(struct task_group *tg) {} 84 static inline int scx_cgroup_can_attach(struct cgroup_taskset *tset) { return 0; } 85 static inline void scx_move_task(struct task_struct *p) {} 86 static inline void scx_cgroup_finish_attach(void) {} 87 static inline void scx_cgroup_cancel_attach(struct cgroup_taskset *tset) {} 88 static inline void scx_group_set_weight(struct task_group *tg, unsigned long cgrp_weight) {} 89 static inline void scx_group_set_idle(struct task_group *tg, bool idle) {} 90 #endif /* CONFIG_EXT_GROUP_SCHED */ 91 #endif /* CONFIG_CGROUP_SCHED */ 92