1 files changed, 9 insertions, 7 deletions

diff --git a/arch/arm/common/bL_switcher.c b/arch/arm/common/bL_switcher.c
index 5774b6ea7ad5..490f3dced749 100644
--- a/arch/arm/common/bL_switcher.c
+++ b/arch/arm/common/bL_switcher.c
@@ -433,8 +433,12 @@ static void bL_switcher_restore_cpus(void)
 {
         int i;
 
-        for_each_cpu(i, &bL_switcher_removed_logical_cpus)
-                cpu_up(i);
+        for_each_cpu(i, &bL_switcher_removed_logical_cpus) {
+                struct device *cpu_dev = get_cpu_device(i);
+                int ret = device_online(cpu_dev);
+                if (ret)
+                        dev_err(cpu_dev, "switcher: unable to restore CPU\n");
+        }
 }
 
 static int bL_switcher_halve_cpus(void)
@@ -521,7 +525,7 @@ static int bL_switcher_halve_cpus(void)
                         continue;
                 }
 
-                ret = cpu_down(i);
+                ret = device_offline(get_cpu_device(i));
                 if (ret) {
                         bL_switcher_restore_cpus();
                         return ret;
@@ -797,10 +801,8 @@ static int __init bL_switcher_init(void)
 {
         int ret;
 
-        if (MAX_NR_CLUSTERS != 2) {
-                pr_err("%s: only dual cluster systems are supported\n", __func__);
-                return -EINVAL;
-        }
+        if (!mcpm_is_available())
+                return -ENODEV;
 
         cpu_notifier(bL_switcher_hotplug_callback, 0);