hugetlb_vmemmap.h 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. // SPDX-License-Identifier: GPL-2.0
  2. /*
  3. * HugeTLB Vmemmap Optimization (HVO)
  4. *
  5. * Copyright (c) 2020, ByteDance. All rights reserved.
  6. *
  7. * Author: Muchun Song <[email protected]>
  8. */
  9. #ifndef _LINUX_HUGETLB_VMEMMAP_H
  10. #define _LINUX_HUGETLB_VMEMMAP_H
  11. #include <linux/hugetlb.h>
  12. #ifdef CONFIG_HUGETLB_PAGE_OPTIMIZE_VMEMMAP
  13. int hugetlb_vmemmap_restore(const struct hstate *h, struct page *head);
  14. void hugetlb_vmemmap_optimize(const struct hstate *h, struct page *head);
  15. /*
  16. * Reserve one vmemmap page, all vmemmap addresses are mapped to it. See
  17. * Documentation/vm/vmemmap_dedup.rst.
  18. */
  19. #define HUGETLB_VMEMMAP_RESERVE_SIZE PAGE_SIZE
  20. static inline unsigned int hugetlb_vmemmap_size(const struct hstate *h)
  21. {
  22. return pages_per_huge_page(h) * sizeof(struct page);
  23. }
  24. /*
  25. * Return how many vmemmap size associated with a HugeTLB page that can be
  26. * optimized and can be freed to the buddy allocator.
  27. */
  28. static inline unsigned int hugetlb_vmemmap_optimizable_size(const struct hstate *h)
  29. {
  30. int size = hugetlb_vmemmap_size(h) - HUGETLB_VMEMMAP_RESERVE_SIZE;
  31. if (!is_power_of_2(sizeof(struct page)))
  32. return 0;
  33. return size > 0 ? size : 0;
  34. }
  35. #else
  36. static inline int hugetlb_vmemmap_restore(const struct hstate *h, struct page *head)
  37. {
  38. return 0;
  39. }
  40. static inline void hugetlb_vmemmap_optimize(const struct hstate *h, struct page *head)
  41. {
  42. }
  43. static inline unsigned int hugetlb_vmemmap_optimizable_size(const struct hstate *h)
  44. {
  45. return 0;
  46. }
  47. #endif /* CONFIG_HUGETLB_PAGE_OPTIMIZE_VMEMMAP */
  48. static inline bool hugetlb_vmemmap_optimizable(const struct hstate *h)
  49. {
  50. return hugetlb_vmemmap_optimizable_size(h) != 0;
  51. }
  52. #endif /* _LINUX_HUGETLB_VMEMMAP_H */