1 #ifndef _LINUX_SCHED_ISOLATION_H
2 #define _LINUX_SCHED_ISOLATION_H
3 
4 #include <linux/cpumask.h>
5 #include <linux/init.h>
6 #include <linux/tick.h>
7 
8 enum hk_flags {
9 	HK_FLAG_TIMER		= 1,
10 	HK_FLAG_RCU		= (1 << 1),
11 	HK_FLAG_MISC		= (1 << 2),
12 	HK_FLAG_SCHED		= (1 << 3),
13 	HK_FLAG_TICK		= (1 << 4),
14 	HK_FLAG_DOMAIN		= (1 << 5),
15 	HK_FLAG_WQ		= (1 << 6),
16 };
17 
18 #ifdef CONFIG_CPU_ISOLATION
19 DECLARE_STATIC_KEY_FALSE(housekeeping_overriden);
20 extern int housekeeping_any_cpu(enum hk_flags flags);
21 extern const struct cpumask *housekeeping_cpumask(enum hk_flags flags);
22 extern void housekeeping_affine(struct task_struct *t, enum hk_flags flags);
23 extern bool housekeeping_test_cpu(int cpu, enum hk_flags flags);
24 extern void __init housekeeping_init(void);
25 
26 #else
27 
housekeeping_any_cpu(enum hk_flags flags)28 static inline int housekeeping_any_cpu(enum hk_flags flags)
29 {
30 	return smp_processor_id();
31 }
32 
housekeeping_cpumask(enum hk_flags flags)33 static inline const struct cpumask *housekeeping_cpumask(enum hk_flags flags)
34 {
35 	return cpu_possible_mask;
36 }
37 
housekeeping_affine(struct task_struct * t,enum hk_flags flags)38 static inline void housekeeping_affine(struct task_struct *t,
39 				       enum hk_flags flags) { }
housekeeping_init(void)40 static inline void housekeeping_init(void) { }
41 #endif /* CONFIG_CPU_ISOLATION */
42 
housekeeping_cpu(int cpu,enum hk_flags flags)43 static inline bool housekeeping_cpu(int cpu, enum hk_flags flags)
44 {
45 #ifdef CONFIG_CPU_ISOLATION
46 	if (static_branch_unlikely(&housekeeping_overriden))
47 		return housekeeping_test_cpu(cpu, flags);
48 #endif
49 	return true;
50 }
51 
52 #endif /* _LINUX_SCHED_ISOLATION_H */
53