1 files changed, 3 insertions, 2 deletions

diff --git a/mm/memblock.c b/mm/memblock.c
index 82aa349d2f7a..931eef145af5 100644
--- a/mm/memblock.c
+++ b/mm/memblock.c
@@ -41,7 +41,8 @@ static int memblock_memory_in_slab __initdata_memblock = 0;
 static int memblock_reserved_in_slab __initdata_memblock = 0;
 
 /* inline so we don't get a warning when pr_debug is compiled out */
-static inline const char *memblock_type_name(struct memblock_type *type)
+static __init_memblock const char *
+memblock_type_name(struct memblock_type *type)
 {
         if (type == &memblock.memory)
                 return "memory";
@@ -756,7 +757,7 @@ int __init_memblock memblock_set_node(phys_addr_t base, phys_addr_t size,
                 return ret;
 
         for (i = start_rgn; i < end_rgn; i++)
-                type->regions[i].nid = nid;
+                memblock_set_region_node(&type->regions[i], nid);
 
         memblock_merge_regions(type);
         return 0;