2 files changed, 49 insertions, 21 deletions

diff --git a/arch/ppc64/mm/init.c b/arch/ppc64/mm/init.c
index c02dc9809c..b3b1e9c177 100644
--- a/arch/ppc64/mm/init.c
+++ b/arch/ppc64/mm/init.c
@@ -552,27 +552,18 @@ void __init do_init_bootmem(void)
         /* Add all physical memory to the bootmem map, mark each area
          * present.
          */
-        for (i=0; i < lmb.memory.cnt; i++) {
-                unsigned long base, size;
-                unsigned long start_pfn, end_pfn;
-
-                base = lmb.memory.region[i].base;
-                size = lmb.memory.region[i].size;
-
-                start_pfn = base >> PAGE_SHIFT;
-                end_pfn = start_pfn + (size >> PAGE_SHIFT);
-                memory_present(0, start_pfn, end_pfn);
-
-                free_bootmem(base, size);
-        }
+        for (i=0; i < lmb.memory.cnt; i++)
+                free_bootmem(lmb_start_pfn(&lmb.memory, i),
+                             lmb_size_bytes(&lmb.memory, i));
 
         /* reserve the sections we're already using */
-        for (i=0; i < lmb.reserved.cnt; i++) {
-                unsigned long base = lmb.reserved.region[i].base;
-                unsigned long size = lmb.reserved.region[i].size;
+        for (i=0; i < lmb.reserved.cnt; i++)
+                reserve_bootmem(lmb_start_pfn(&lmb.reserved, i),
+                                lmb_size_bytes(&lmb.reserved, i));
 
-                reserve_bootmem(base, size);
-        }
+        for (i=0; i < lmb.memory.cnt; i++)
+                memory_present(0, lmb_start_pfn(&lmb.memory, i),
+                               lmb_end_pfn(&lmb.memory, i));
 }
 
 /*
diff --git a/arch/ppc64/mm/numa.c b/arch/ppc64/mm/numa.c
index c3116f0d78..cb864b8f27 100644
--- a/arch/ppc64/mm/numa.c
+++ b/arch/ppc64/mm/numa.c
@@ -440,8 +440,6 @@ new_range:
                 for (i = start ; i < (start+size); i += MEMORY_INCREMENT)
                         numa_memory_lookup_table[i >> MEMORY_INCREMENT_SHIFT] =
                                 numa_domain;
-                memory_present(numa_domain, start >> PAGE_SHIFT,
-                                                (start + size) >> PAGE_SHIFT);
 
                 if (--ranges)
                         goto new_range;
@@ -483,7 +481,6 @@ static void __init setup_nonnuma(void)
 
         for (i = 0 ; i < top_of_ram; i += MEMORY_INCREMENT)
                 numa_memory_lookup_table[i >> MEMORY_INCREMENT_SHIFT] = 0;
-        memory_present(0, 0, init_node_data[0].node_end_pfn);
 }
 
 static void __init dump_numa_topology(void)
@@ -695,6 +692,46 @@ new_range:
                                                      size);
                         }
                 }
+                /*
+                 * This loop may look famaliar, but we have to do it again
+                 * after marking our reserved memory to mark memory present
+                 * for sparsemem.
+                 */
+                addr_cells = get_mem_addr_cells();
+                size_cells = get_mem_size_cells();
+                memory = NULL;
+                while ((memory = of_find_node_by_type(memory, "memory")) != NULL) {
+                        unsigned long mem_start, mem_size;
+                        int numa_domain, ranges;
+                        unsigned int *memcell_buf;
+                        unsigned int len;
+
+                        memcell_buf = (unsigned int *)get_property(memory, "reg", &len);
+                        if (!memcell_buf || len <= 0)
+                                continue;
+
+                        ranges = memory->n_addrs;       /* ranges in cell */
+new_range2:
+                        mem_start = read_n_cells(addr_cells, &memcell_buf);
+                        mem_size = read_n_cells(size_cells, &memcell_buf);
+                        if (numa_enabled) {
+                                numa_domain = of_node_numa_domain(memory);
+                                if (numa_domain  >= MAX_NUMNODES)
+                                        numa_domain = 0;
+                        } else
+                                numa_domain =  0;
+
+                        if (numa_domain != nid)
+                                continue;
+
+                        mem_size = numa_enforce_memory_limit(mem_start, mem_size);
+                        memory_present(numa_domain, mem_start >> PAGE_SHIFT,
+                                       (mem_start + mem_size) >> PAGE_SHIFT);
+
+                        if (--ranges)           /* process all ranges in cell */
+                                goto new_range2;
+                }
+
         }
 }