1 files changed, 5 insertions, 0 deletions

diff --git a/drivers/cpuidle/cpuidle-powernv.c b/drivers/cpuidle/cpuidle-powernv.c
index 78fd174c57e8..f48607cd2540 100644
--- a/drivers/cpuidle/cpuidle-powernv.c
+++ b/drivers/cpuidle/cpuidle-powernv.c
@@ -14,6 +14,7 @@
 
 #include <asm/machdep.h>
 #include <asm/firmware.h>
+#include <asm/runlatch.h>
 
 struct cpuidle_driver powernv_idle_driver = {
         .name             = "powernv_idle",
@@ -30,12 +31,14 @@ static int snooze_loop(struct cpuidle_device *dev,
         local_irq_enable();
         set_thread_flag(TIF_POLLING_NRFLAG);
 
+        ppc64_runlatch_off();
         while (!need_resched()) {
                 HMT_low();
                 HMT_very_low();
         }
 
         HMT_medium();
+        ppc64_runlatch_on();
         clear_thread_flag(TIF_POLLING_NRFLAG);
         smp_mb();
         return index;
@@ -45,7 +48,9 @@ static int nap_loop(struct cpuidle_device *dev,
                         struct cpuidle_driver *drv,
                         int index)
 {
+        ppc64_runlatch_off();
         power7_idle();
+        ppc64_runlatch_on();
         return index;
 }