1 files changed, 7 insertions, 7 deletions

diff --git a/drivers/hwmon/coretemp.c b/drivers/hwmon/coretemp.c
index f31bc4c48644..6d02e3b06375 100644
--- a/drivers/hwmon/coretemp.c
+++ b/drivers/hwmon/coretemp.c
@@ -810,20 +810,20 @@ static int __init coretemp_init(void)
         if (err)
                 goto exit;
 
-        get_online_cpus();
+        cpu_notifier_register_begin();
         for_each_online_cpu(i)
                 get_core_online(i);
 
 #ifndef CONFIG_HOTPLUG_CPU
         if (list_empty(&pdev_list)) {
-                put_online_cpus();
+                cpu_notifier_register_done();
                 err = -ENODEV;
                 goto exit_driver_unreg;
         }
 #endif
 
-        register_hotcpu_notifier(&coretemp_cpu_notifier);
-        put_online_cpus();
+        __register_hotcpu_notifier(&coretemp_cpu_notifier);
+        cpu_notifier_register_done();
         return 0;
 
 #ifndef CONFIG_HOTPLUG_CPU
@@ -838,8 +838,8 @@ static void __exit coretemp_exit(void)
 {
         struct pdev_entry *p, *n;
 
-        get_online_cpus();
-        unregister_hotcpu_notifier(&coretemp_cpu_notifier);
+        cpu_notifier_register_begin();
+        __unregister_hotcpu_notifier(&coretemp_cpu_notifier);
         mutex_lock(&pdev_list_mutex);
         list_for_each_entry_safe(p, n, &pdev_list, list) {
                 platform_device_unregister(p->pdev);
@@ -847,7 +847,7 @@ static void __exit coretemp_exit(void)
                 kfree(p);
         }
         mutex_unlock(&pdev_list_mutex);
-        put_online_cpus();
+        cpu_notifier_register_done();
         platform_driver_unregister(&coretemp_driver);
 }