1 files changed, 3 insertions, 3 deletions

diff --git a/arch/x86/kernel/cpu/perf_event_intel_uncore.c b/arch/x86/kernel/cpu/perf_event_intel_uncore.c
index 047f540cf3f7..28922f62eb29 100644
--- a/arch/x86/kernel/cpu/perf_event_intel_uncore.c
+++ b/arch/x86/kernel/cpu/perf_event_intel_uncore.c
@@ -3817,7 +3817,7 @@ static int __init uncore_cpu_init(void)
         if (ret)
                 return ret;
 
-        get_online_cpus();
+        cpu_notifier_register_begin();
 
         for_each_online_cpu(cpu) {
                 int i, phys_id = topology_physical_package_id(cpu);
@@ -3836,9 +3836,9 @@ static int __init uncore_cpu_init(void)
         }
         on_each_cpu(uncore_cpu_setup, NULL, 1);
 
-        register_cpu_notifier(&uncore_cpu_nb);
+        __register_cpu_notifier(&uncore_cpu_nb);
 
-        put_online_cpus();
+        cpu_notifier_register_done();
 
         return 0;
 }