1 files changed, 9 insertions, 11 deletions

diff --git a/arch/microblaze/mm/init.c b/arch/microblaze/mm/init.c
index afd6494fbbc6..32a702b621c8 100644
--- a/arch/microblaze/mm/init.c
+++ b/arch/microblaze/mm/init.c
@@ -70,16 +70,16 @@ static void __init paging_init(void)
 
 void __init setup_memory(void)
 {
-        int i;
         unsigned long map_size;
+        struct memblock_region *reg;
+
 #ifndef CONFIG_MMU
         u32 kernel_align_start, kernel_align_size;
 
         /* Find main memory where is the kernel */
-        for (i = 0; i < memblock.memory.cnt; i++) {
-                memory_start = (u32) memblock.memory.regions[i].base;
-                memory_end = (u32) memblock.memory.regions[i].base
-                                + (u32) memblock.memory.region[i].size;
+        for_each_memblock(memory, reg) {
+                memory_start = (u32)reg->base;
+                memory_end = (u32) reg->base + reg->size;
                 if ((memory_start <= (u32)_text) &&
                                         ((u32)_text <= memory_end)) {
                         memory_size = memory_end - memory_start;
@@ -147,12 +147,10 @@ void __init setup_memory(void)
         free_bootmem(memory_start, memory_size);
 
         /* reserve allocate blocks */
-        for (i = 0; i < memblock.reserved.cnt; i++) {
-                pr_debug("reserved %d - 0x%08x-0x%08x\n", i,
-                        (u32) memblock.reserved.region[i].base,
-                        (u32) memblock_size_bytes(&memblock.reserved, i));
-                reserve_bootmem(memblock.reserved.region[i].base,
-                        memblock_size_bytes(&memblock.reserved, i) - 1, BOOTMEM_DEFAULT);
+        for_each_memblock(reserved, reg) {
+                pr_debug("reserved - 0x%08x-0x%08x\n",
+                         (u32) reg->base, (u32) reg->size);
+                reserve_bootmem(reg->base, reg->size, BOOTMEM_DEFAULT);
         }
 #ifdef CONFIG_MMU
         init_bootmem_done = 1;