struct freq_attr *fattr = to_attr(attr);
        ssize_t ret = -EINVAL;
 
-       get_online_cpus();
+       cpus_read_lock();
 
        if (cpu_online(policy->cpu)) {
                down_write(&policy->rwsem);
                up_write(&policy->rwsem);
        }
 
-       put_online_cpus();
+       cpus_read_unlock();
 
        return ret;
 }
        pr_debug("trying to register driver %s\n", driver_data->name);
 
        /* Protect against concurrent CPU online/offline. */
-       get_online_cpus();
+       cpus_read_lock();
 
        write_lock_irqsave(&cpufreq_driver_lock, flags);
        if (cpufreq_driver) {
                goto err_if_unreg;
        }
 
-       ret = cpuhp_setup_state_nocalls(CPUHP_AP_ONLINE_DYN, "cpufreq:online",
-                                       cpuhp_cpufreq_online,
-                                       cpuhp_cpufreq_offline);
+       ret = cpuhp_setup_state_nocalls_cpuslocked(CPUHP_AP_ONLINE_DYN,
+                                                  "cpufreq:online",
+                                                  cpuhp_cpufreq_online,
+                                                  cpuhp_cpufreq_offline);
        if (ret < 0)
                goto err_if_unreg;
        hp_online = ret;
        cpufreq_driver = NULL;
        write_unlock_irqrestore(&cpufreq_driver_lock, flags);
 out:
-       put_online_cpus();
+       cpus_read_unlock();
        return ret;
 }
 EXPORT_SYMBOL_GPL(cpufreq_register_driver);
        pr_debug("unregistering driver %s\n", driver->name);
 
        /* Protect against concurrent cpu hotplug */
-       get_online_cpus();
+       cpus_read_lock();
        subsys_interface_unregister(&cpufreq_interface);
        remove_boost_sysfs_file();
-       cpuhp_remove_state_nocalls(hp_online);
+       cpuhp_remove_state_nocalls_cpuslocked(hp_online);
 
        write_lock_irqsave(&cpufreq_driver_lock, flags);
 
        cpufreq_driver = NULL;
 
        write_unlock_irqrestore(&cpufreq_driver_lock, flags);
-       put_online_cpus();
+       cpus_read_unlock();
 
        return 0;
 }