1 files changed, 11 insertions, 1 deletions

diff --git a/mm/sparse-vmemmap.c b/mm/sparse-vmemmap.c
index 22620f6a976b..cd75b21dd4c3 100644
--- a/mm/sparse-vmemmap.c
+++ b/mm/sparse-vmemmap.c
@@ -34,6 +34,16 @@
  * or to back the page tables that are used to create the mapping.
  * Uses the main allocators if they are available, else bootmem.
  */
+
+static void * __init_refok __earlyonly_bootmem_alloc(int node,
+                                unsigned long size,
+                                unsigned long align,
+                                unsigned long goal)
+{
+        return __alloc_bootmem_node(NODE_DATA(node), size, align, goal);
+}
+
+
 void * __meminit vmemmap_alloc_block(unsigned long size, int node)
 {
         /* If the main allocator is up use that, fallback to bootmem. */
@@ -44,7 +54,7 @@ void * __meminit vmemmap_alloc_block(unsigned long size, int node)
                         return page_address(page);
                 return NULL;
         } else
-                return __alloc_bootmem_node(NODE_DATA(node), size, size,
+                return __earlyonly_bootmem_alloc(node, size, size,
                                 __pa(MAX_DMA_ADDRESS));
 }