1 files changed, 2 insertions, 1 deletions

diff --git a/arch/x86/mm/numa_64.c b/arch/x86/mm/numa_64.c
index cccc01d8415c..7757d2214fab 100644
--- a/arch/x86/mm/numa_64.c
+++ b/arch/x86/mm/numa_64.c
@@ -414,7 +414,8 @@ static int __init numa_alloc_distance(void)
 
         for_each_node_mask(i, nodes_parsed)
                 cnt = i;
-        size = ++cnt * sizeof(numa_distance[0]);
+        cnt++;
+        size = cnt * cnt * sizeof(numa_distance[0]);
 
         phys = memblock_find_in_range(0, (u64)max_pfn_mapped << PAGE_SHIFT,
                                       size, PAGE_SIZE);