On Fri, Nov 13, 2020 at 06:59:36PM +0800, Muchun Song wrote:
> +#define page_huge_pte(page)          ((page)->pmd_huge_pte)

Seems you do not need this one anymore.

> +void vmemmap_pgtable_free(struct page *page)
> +{
> +     struct page *pte_page, *t_page;
> +
> +     list_for_each_entry_safe(pte_page, t_page, &page->lru, lru) {
> +             list_del(&pte_page->lru);
> +             pte_free_kernel(&init_mm, page_to_virt(pte_page));
> +     }
> +}
> +
> +int vmemmap_pgtable_prealloc(struct hstate *h, struct page *page)
> +{
> +     unsigned int nr = pgtable_pages_to_prealloc_per_hpage(h);
> +
> +     /* Store preallocated pages on huge page lru list */
> +     INIT_LIST_HEAD(&page->lru);
> +
> +     while (nr--) {
> +             pte_t *pte_p;
> +
> +             pte_p = pte_alloc_one_kernel(&init_mm);
> +             if (!pte_p)
> +                     goto out;
> +             list_add(&virt_to_page(pte_p)->lru, &page->lru);
> +     }

Definetely this looks better and easier to handle.
Btw, did you explore Matthew's hint about instead of allocating a new page,
using one of the ones you are going to free to store the ptes?
I am not sure whether it is feasible at all though.


> --- a/mm/hugetlb_vmemmap.h
> +++ b/mm/hugetlb_vmemmap.h
> @@ -9,12 +9,24 @@
>  #ifndef _LINUX_HUGETLB_VMEMMAP_H
>  #define _LINUX_HUGETLB_VMEMMAP_H
>  #include <linux/hugetlb.h>
> +#include <linux/mm.h>

why do we need this here?

-- 
Oscar Salvador
SUSE L3

Reply via email to