void __init allocate_pgdat(unsigned int nid)
 {
        unsigned long start_pfn, end_pfn;
-#ifdef CONFIG_NEED_MULTIPLE_NODES
-       unsigned long phys;
-#endif
 
        get_pfn_range_for_nid(nid, &start_pfn, &end_pfn);
 
 #ifdef CONFIG_NEED_MULTIPLE_NODES
-       phys = __memblock_alloc_base(sizeof(struct pglist_data),
-                               SMP_CACHE_BYTES, end_pfn << PAGE_SHIFT);
-       /* Retry with all of system memory */
-       if (!phys)
-               phys = __memblock_alloc_base(sizeof(struct pglist_data),
-                                       SMP_CACHE_BYTES, memblock_end_of_DRAM());
-       if (!phys)
+       NODE_DATA(nid) = memblock_alloc_try_nid_nopanic(
+                               sizeof(struct pglist_data),
+                               SMP_CACHE_BYTES, MEMBLOCK_LOW_LIMIT,
+                               MEMBLOCK_ALLOC_ACCESSIBLE, nid);
+       if (!NODE_DATA(nid))
                panic("Can't allocate pgdat for node %d\n", nid);
-
-       NODE_DATA(nid) = __va(phys);
-       memset(NODE_DATA(nid), 0, sizeof(struct pglist_data));
 #endif
 
        NODE_DATA(nid)->node_start_pfn = start_pfn;
 
        __add_active_range(nid, start_pfn, end_pfn);
 
        /* Node-local pgdat */
-       NODE_DATA(nid) = __va(memblock_alloc_base(sizeof(struct pglist_data),
-                                            SMP_CACHE_BYTES, end));
-       memset(NODE_DATA(nid), 0, sizeof(struct pglist_data));
+       NODE_DATA(nid) = memblock_alloc_node(sizeof(struct pglist_data),
+                                            SMP_CACHE_BYTES, nid);
 
        NODE_DATA(nid)->node_start_pfn = start_pfn;
        NODE_DATA(nid)->node_spanned_pages = end_pfn - start_pfn;