psi.h 1.7 KB

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