1 files changed, 2 insertions, 2 deletions

diff --git a/arch/x86/kernel/nmi_64.c b/arch/x86/kernel/nmi_64.c
index 0060e44e8989..d62f3b66b529 100644
--- a/arch/x86/kernel/nmi_64.c
+++ b/arch/x86/kernel/nmi_64.c
@@ -90,7 +90,7 @@ int __init check_nmi_watchdog(void)
         if (!atomic_read(&nmi_active))
                 return 0;
 
-        prev_nmi_count = kmalloc(NR_CPUS * sizeof(int), GFP_KERNEL);
+        prev_nmi_count = kmalloc(nr_cpu_ids * sizeof(int), GFP_KERNEL);
         if (!prev_nmi_count)
                 goto error;
 
@@ -101,7 +101,7 @@ int __init check_nmi_watchdog(void)
                 smp_call_function(nmi_cpu_busy, (void *)&endflag, 0, 0);
 #endif
 
-        for (cpu = 0; cpu < NR_CPUS; cpu++)
+        for (cpu = 0; cpu < nr_cpu_ids; cpu++)
                 prev_nmi_count[cpu] = cpu_pda(cpu)->__nmi_count;
         local_irq_enable();
         mdelay((20*1000)/nmi_hz); // wait 20 ticks