VMEMMAP_POPULATE_PAGEREF is only needed once the slab allocator is
available, so it does not need to be passed through the vmemmap
population call chain.

Drop the flag and test slab_is_available() directly in
vmemmap_pte_populate() instead.

Signed-off-by: Muchun Song <[email protected]>
---
 mm/sparse-vmemmap.c | 40 ++++++++++++++--------------------------
 1 file changed, 14 insertions(+), 26 deletions(-)

diff --git a/mm/sparse-vmemmap.c b/mm/sparse-vmemmap.c
index 17d45dac4324..d7e9fb47f7ee 100644
--- a/mm/sparse-vmemmap.c
+++ b/mm/sparse-vmemmap.c
@@ -33,13 +33,6 @@
 #include <asm/tlbflush.h>
 
 #include "hugetlb_vmemmap.h"
-
-/*
- * Flags for vmemmap_populate_range and friends.
- */
-/* Get a ref on the head page struct page, for ZONE_DEVICE compound pages */
-#define VMEMMAP_POPULATE_PAGEREF       0x0001
-
 #include "internal.h"
 
 /*
@@ -147,8 +140,8 @@ void __meminit vmemmap_verify(pte_t *pte, int node,
 }
 
 static pte_t * __meminit vmemmap_pte_populate(pmd_t *pmd, unsigned long addr, 
int node,
-                                      struct vmem_altmap *altmap,
-                                      unsigned long ptpfn, unsigned long flags)
+                                             struct vmem_altmap *altmap,
+                                             unsigned long ptpfn)
 {
        pte_t *pte = pte_offset_kernel(pmd, addr);
        if (pte_none(ptep_get(pte))) {
@@ -170,7 +163,7 @@ static pte_t * __meminit vmemmap_pte_populate(pmd_t *pmd, 
unsigned long addr, in
                         * and through vmemmap_populate_compound_pages() when
                         * slab is available.
                         */
-                       if (flags & VMEMMAP_POPULATE_PAGEREF)
+                       if (slab_is_available())
                                get_page(pfn_to_page(ptpfn));
                }
                entry = pfn_pte(ptpfn, PAGE_KERNEL);
@@ -243,8 +236,7 @@ static pgd_t * __meminit vmemmap_pgd_populate(unsigned long 
addr, int node)
 
 static pte_t * __meminit vmemmap_populate_address(unsigned long addr, int node,
                                              struct vmem_altmap *altmap,
-                                             unsigned long ptpfn,
-                                             unsigned long flags)
+                                             unsigned long ptpfn)
 {
        pgd_t *pgd;
        p4d_t *p4d;
@@ -264,7 +256,7 @@ static pte_t * __meminit vmemmap_populate_address(unsigned 
long addr, int node,
        pmd = vmemmap_pmd_populate(pud, addr, node);
        if (!pmd)
                return NULL;
-       pte = vmemmap_pte_populate(pmd, addr, node, altmap, ptpfn, flags);
+       pte = vmemmap_pte_populate(pmd, addr, node, altmap, ptpfn);
        if (!pte)
                return NULL;
        vmemmap_verify(pte, node, addr, addr + PAGE_SIZE);
@@ -275,15 +267,14 @@ static pte_t * __meminit 
vmemmap_populate_address(unsigned long addr, int node,
 static int __meminit vmemmap_populate_range(unsigned long start,
                                            unsigned long end, int node,
                                            struct vmem_altmap *altmap,
-                                           unsigned long ptpfn,
-                                           unsigned long flags)
+                                           unsigned long ptpfn)
 {
        unsigned long addr = start;
        pte_t *pte;
 
        for (; addr < end; addr += PAGE_SIZE) {
                pte = vmemmap_populate_address(addr, node, altmap,
-                                              ptpfn, flags);
+                                              ptpfn);
                if (!pte)
                        return -ENOMEM;
        }
@@ -294,7 +285,7 @@ static int __meminit vmemmap_populate_range(unsigned long 
start,
 int __meminit vmemmap_populate_basepages(unsigned long start, unsigned long 
end,
                                         int node, struct vmem_altmap *altmap)
 {
-       return vmemmap_populate_range(start, end, node, altmap, -1, 0);
+       return vmemmap_populate_range(start, end, node, altmap, -1);
 }
 
 /*
@@ -370,7 +361,7 @@ int __meminit vmemmap_populate_hvo(unsigned long addr, 
unsigned long end,
                return -ENOMEM;
 
        for (maddr = addr; maddr < addr + headsize; maddr += PAGE_SIZE) {
-               pte = vmemmap_populate_address(maddr, node, NULL, -1, 0);
+               pte = vmemmap_populate_address(maddr, node, NULL, -1);
                if (!pte)
                        return -ENOMEM;
        }
@@ -378,8 +369,7 @@ int __meminit vmemmap_populate_hvo(unsigned long addr, 
unsigned long end,
        /*
         * Reuse the last page struct page mapped above for the rest.
         */
-       return vmemmap_populate_range(maddr, end, node, NULL,
-                                     page_to_pfn(tail), 0);
+       return vmemmap_populate_range(maddr, end, node, NULL, 
page_to_pfn(tail));
 }
 #endif
 
@@ -503,8 +493,7 @@ static int __meminit 
vmemmap_populate_compound_pages(unsigned long start_pfn,
                 * with just tail struct pages.
                 */
                return vmemmap_populate_range(start, end, node, NULL,
-                                             pte_pfn(ptep_get(pte)),
-                                             VMEMMAP_POPULATE_PAGEREF);
+                                             pte_pfn(ptep_get(pte)));
        }
 
        size = min(end - start, pgmap_vmemmap_nr(pgmap) * sizeof(struct page));
@@ -512,13 +501,13 @@ static int __meminit 
vmemmap_populate_compound_pages(unsigned long start_pfn,
                unsigned long next, last = addr + size;
 
                /* Populate the head page vmemmap page */
-               pte = vmemmap_populate_address(addr, node, NULL, -1, 0);
+               pte = vmemmap_populate_address(addr, node, NULL, -1);
                if (!pte)
                        return -ENOMEM;
 
                /* Populate the tail pages vmemmap page */
                next = addr + PAGE_SIZE;
-               pte = vmemmap_populate_address(next, node, NULL, -1, 0);
+               pte = vmemmap_populate_address(next, node, NULL, -1);
                if (!pte)
                        return -ENOMEM;
 
@@ -528,8 +517,7 @@ static int __meminit 
vmemmap_populate_compound_pages(unsigned long start_pfn,
                 */
                next += PAGE_SIZE;
                rc = vmemmap_populate_range(next, last, node, NULL,
-                                           pte_pfn(ptep_get(pte)),
-                                           VMEMMAP_POPULATE_PAGEREF);
+                                           pte_pfn(ptep_get(pte)));
                if (rc)
                        return -ENOMEM;
        }
-- 
2.54.0


Reply via email to