1 files changed, 11 insertions, 7 deletions

diff --git a/arch/sh/kernel/cpu/shmobile/cpuidle.c b/arch/sh/kernel/cpu/shmobile/cpuidle.c
index 7be50d4c4268..ad1012ad6b42 100644
--- a/arch/sh/kernel/cpu/shmobile/cpuidle.c
+++ b/arch/sh/kernel/cpu/shmobile/cpuidle.c
@@ -25,6 +25,7 @@ static unsigned long cpuidle_mode[] = {
 };
 
 static int cpuidle_sleep_enter(struct cpuidle_device *dev,
+                                struct cpuidle_driver *drv,
                                 int index)
 {
         unsigned long allowed_mode = arch_hwblk_sleep_mode();
@@ -64,19 +65,19 @@ static struct cpuidle_driver cpuidle_driver = {
 void sh_mobile_setup_cpuidle(void)
 {
         struct cpuidle_device *dev = &cpuidle_dev;
+        struct cpuidle_driver *drv = &cpuidle_driver;
         struct cpuidle_state *state;
         int i;
 
-        cpuidle_register_driver(&cpuidle_driver);
 
         for (i = 0; i < CPUIDLE_STATE_MAX; i++) {
-                dev->states[i].name[0] = '\0';
-                dev->states[i].desc[0] = '\0';
+                drv->states[i].name[0] = '\0';
+                drv->states[i].desc[0] = '\0';
         }
 
         i = CPUIDLE_DRIVER_STATE_START;
 
-        state = &dev->states[i++];
+        state = &drv->states[i++];
         snprintf(state->name, CPUIDLE_NAME_LEN, "C1");
         strncpy(state->desc, "SuperH Sleep Mode", CPUIDLE_DESC_LEN);
         state->exit_latency = 1;
@@ -86,10 +87,10 @@ void sh_mobile_setup_cpuidle(void)
         state->flags |= CPUIDLE_FLAG_TIME_VALID;
         state->enter = cpuidle_sleep_enter;
 
-        dev->safe_state_index = i-1;
+        drv->safe_state_index = i-1;
 
         if (sh_mobile_sleep_supported & SUSP_SH_SF) {
-                state = &dev->states[i++];
+                state = &drv->states[i++];
                 snprintf(state->name, CPUIDLE_NAME_LEN, "C2");
                 strncpy(state->desc, "SuperH Sleep Mode [SF]",
                         CPUIDLE_DESC_LEN);
@@ -102,7 +103,7 @@ void sh_mobile_setup_cpuidle(void)
         }
 
         if (sh_mobile_sleep_supported & SUSP_SH_STANDBY) {
-                state = &dev->states[i++];
+                state = &drv->states[i++];
                 snprintf(state->name, CPUIDLE_NAME_LEN, "C3");
                 strncpy(state->desc, "SuperH Mobile Standby Mode [SF]",
                         CPUIDLE_DESC_LEN);
@@ -114,7 +115,10 @@ void sh_mobile_setup_cpuidle(void)
                 state->enter = cpuidle_sleep_enter;
         }
 
+        drv->state_count = i;
         dev->state_count = i;
 
+        cpuidle_register_driver(&cpuidle_driver);
+
         cpuidle_register_device(dev);
 }