1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef _LINUX_PSI_H 3 #define _LINUX_PSI_H 4 5 #include <linux/jump_label.h> 6 #include <linux/psi_types.h> 7 #include <linux/sched.h> 8 #include <linux/poll.h> 9 #include <linux/cgroup-defs.h> 10 #include <linux/cgroup.h> 11 12 struct seq_file; 13 struct css_set; 14 15 #ifdef CONFIG_PSI 16 17 extern struct static_key_false psi_disabled; 18 extern struct psi_group psi_system; 19 20 void psi_init(void); 21 22 void psi_memstall_enter(unsigned long *flags); 23 void psi_memstall_leave(unsigned long *flags); 24 25 int psi_show(struct seq_file *s, struct psi_group *group, enum psi_res res); 26 struct psi_trigger *psi_trigger_create(struct psi_group *group, char *buf, 27 enum psi_res res, struct file *file, 28 struct kernfs_open_file *of); 29 void psi_trigger_destroy(struct psi_trigger *t); 30 31 __poll_t psi_trigger_poll(void **trigger_ptr, struct file *file, 32 poll_table *wait); 33 34 #ifdef CONFIG_CGROUPS 35 static inline struct psi_group *cgroup_psi(struct cgroup *cgrp) 36 { 37 return cgroup_ino(cgrp) == 1 ? &psi_system : cgrp->psi; 38 } 39 40 int psi_cgroup_alloc(struct cgroup *cgrp); 41 void psi_cgroup_free(struct cgroup *cgrp); 42 void cgroup_move_task(struct task_struct *p, struct css_set *to); 43 void psi_cgroup_restart(struct psi_group *group); 44 #endif 45 46 #else /* CONFIG_PSI */ 47 48 static inline void psi_init(void) {} 49 50 static inline void psi_memstall_enter(unsigned long *flags) {} 51 static inline void psi_memstall_leave(unsigned long *flags) {} 52 53 #ifdef CONFIG_CGROUPS 54 static inline int psi_cgroup_alloc(struct cgroup *cgrp) 55 { 56 return 0; 57 } 58 static inline void psi_cgroup_free(struct cgroup *cgrp) 59 { 60 } 61 static inline void cgroup_move_task(struct task_struct *p, struct css_set *to) 62 { 63 rcu_assign_pointer(p->cgroups, to); 64 } 65 static inline void psi_cgroup_restart(struct psi_group *group) {} 66 #endif 67 68 #endif /* CONFIG_PSI */ 69 70 #endif /* _LINUX_PSI_H */ 71