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