xref: /linux-6.15/include/linux/sched/isolation.h (revision 9f0ca2d9)
178634061SFrederic Weisbecker #ifndef _LINUX_SCHED_ISOLATION_H
278634061SFrederic Weisbecker #define _LINUX_SCHED_ISOLATION_H
378634061SFrederic Weisbecker 
478634061SFrederic Weisbecker #include <linux/cpumask.h>
578634061SFrederic Weisbecker #include <linux/init.h>
678634061SFrederic Weisbecker #include <linux/tick.h>
778634061SFrederic Weisbecker 
878634061SFrederic Weisbecker #ifdef CONFIG_NO_HZ_FULL
978634061SFrederic Weisbecker extern cpumask_var_t housekeeping_mask;
1078634061SFrederic Weisbecker extern void __init housekeeping_init(void);
1178634061SFrederic Weisbecker #else
1278634061SFrederic Weisbecker static inline void housekeeping_init(void) { }
1378634061SFrederic Weisbecker #endif /* CONFIG_NO_HZ_FULL */
1478634061SFrederic Weisbecker 
15*9f0ca2d9SFrederic Weisbecker static inline int housekeeping_any_cpu(void)
16*9f0ca2d9SFrederic Weisbecker {
17*9f0ca2d9SFrederic Weisbecker #ifdef CONFIG_NO_HZ_FULL
18*9f0ca2d9SFrederic Weisbecker 	if (tick_nohz_full_enabled())
19*9f0ca2d9SFrederic Weisbecker 		return cpumask_any_and(housekeeping_mask, cpu_online_mask);
20*9f0ca2d9SFrederic Weisbecker #endif
21*9f0ca2d9SFrederic Weisbecker 	return smp_processor_id();
22*9f0ca2d9SFrederic Weisbecker }
2378634061SFrederic Weisbecker 
2478634061SFrederic Weisbecker static inline const struct cpumask *housekeeping_cpumask(void)
2578634061SFrederic Weisbecker {
2678634061SFrederic Weisbecker #ifdef CONFIG_NO_HZ_FULL
2778634061SFrederic Weisbecker 	if (tick_nohz_full_enabled())
2878634061SFrederic Weisbecker 		return housekeeping_mask;
2978634061SFrederic Weisbecker #endif
3078634061SFrederic Weisbecker 	return cpu_possible_mask;
3178634061SFrederic Weisbecker }
3278634061SFrederic Weisbecker 
3378634061SFrederic Weisbecker static inline bool is_housekeeping_cpu(int cpu)
3478634061SFrederic Weisbecker {
3578634061SFrederic Weisbecker #ifdef CONFIG_NO_HZ_FULL
3678634061SFrederic Weisbecker 	if (tick_nohz_full_enabled())
3778634061SFrederic Weisbecker 		return cpumask_test_cpu(cpu, housekeeping_mask);
3878634061SFrederic Weisbecker #endif
3978634061SFrederic Weisbecker 	return true;
4078634061SFrederic Weisbecker }
4178634061SFrederic Weisbecker 
4278634061SFrederic Weisbecker static inline void housekeeping_affine(struct task_struct *t)
4378634061SFrederic Weisbecker {
4478634061SFrederic Weisbecker #ifdef CONFIG_NO_HZ_FULL
4578634061SFrederic Weisbecker 	if (tick_nohz_full_enabled())
4678634061SFrederic Weisbecker 		set_cpus_allowed_ptr(t, housekeeping_mask);
4778634061SFrederic Weisbecker 
4878634061SFrederic Weisbecker #endif
4978634061SFrederic Weisbecker }
5078634061SFrederic Weisbecker 
5178634061SFrederic Weisbecker #endif /* _LINUX_SCHED_ISOLATION_H */
52