1 files changed, 50 insertions, 4 deletions

diff --git a/drivers/clocksource/sh_tmu.c b/drivers/clocksource/sh_tmu.c
index c1b51d49d106..7d700829bb41 100644
--- a/drivers/clocksource/sh_tmu.c
+++ b/drivers/clocksource/sh_tmu.c
@@ -33,6 +33,7 @@
 #include <linux/slab.h>
 #include <linux/module.h>
 #include <linux/pm_domain.h>
+#include <linux/pm_runtime.h>
 
 struct sh_tmu_priv {
         void __iomem *mapbase;
@@ -43,6 +44,7 @@ struct sh_tmu_priv {
         unsigned long periodic;
         struct clock_event_device ced;
         struct clocksource cs;
+        bool cs_enabled;
 };
 
 static DEFINE_RAW_SPINLOCK(sh_tmu_lock);
@@ -204,14 +206,40 @@ static int sh_tmu_clocksource_enable(struct clocksource *cs)
         int ret;
 
         ret = sh_tmu_enable(p);
-        if (!ret)
+        if (!ret) {
                 __clocksource_updatefreq_hz(cs, p->rate);
+                p->cs_enabled = true;
+        }
         return ret;
 }
 
 static void sh_tmu_clocksource_disable(struct clocksource *cs)
 {
-        sh_tmu_disable(cs_to_sh_tmu(cs));
+        struct sh_tmu_priv *p = cs_to_sh_tmu(cs);
+
+        WARN_ON(!p->cs_enabled);
+
+        sh_tmu_disable(p);
+        p->cs_enabled = false;
+}
+
+static void sh_tmu_clocksource_suspend(struct clocksource *cs)
+{
+        struct sh_tmu_priv *p = cs_to_sh_tmu(cs);
+
+        if (p->cs_enabled)
+                sh_tmu_disable(p);
+
+        pm_genpd_syscore_poweroff(&p->pdev->dev);
+}
+
+static void sh_tmu_clocksource_resume(struct clocksource *cs)
+{
+        struct sh_tmu_priv *p = cs_to_sh_tmu(cs);
+
+        pm_genpd_syscore_poweron(&p->pdev->dev);
+        if (p->cs_enabled)
+                sh_tmu_enable(p);
 }
 
 static int sh_tmu_register_clocksource(struct sh_tmu_priv *p,
@@ -224,6 +252,8 @@ static int sh_tmu_register_clocksource(struct sh_tmu_priv *p,
         cs->read = sh_tmu_clocksource_read;
         cs->enable = sh_tmu_clocksource_enable;
         cs->disable = sh_tmu_clocksource_disable;
+        cs->suspend = sh_tmu_clocksource_suspend;
+        cs->resume = sh_tmu_clocksource_resume;
         cs->mask = CLOCKSOURCE_MASK(32);
         cs->flags = CLOCK_SOURCE_IS_CONTINUOUS;
 
@@ -301,6 +331,16 @@ static int sh_tmu_clock_event_next(unsigned long delta,
         return 0;
 }
 
+static void sh_tmu_clock_event_suspend(struct clock_event_device *ced)
+{
+        pm_genpd_syscore_poweroff(&ced_to_sh_tmu(ced)->pdev->dev);
+}
+
+static void sh_tmu_clock_event_resume(struct clock_event_device *ced)
+{
+        pm_genpd_syscore_poweron(&ced_to_sh_tmu(ced)->pdev->dev);
+}
+
 static void sh_tmu_register_clockevent(struct sh_tmu_priv *p,
                                        char *name, unsigned long rating)
 {
@@ -316,6 +356,8 @@ static void sh_tmu_register_clockevent(struct sh_tmu_priv *p,
         ced->cpumask = cpumask_of(0);
         ced->set_next_event = sh_tmu_clock_event_next;
         ced->set_mode = sh_tmu_clock_event_mode;
+        ced->suspend = sh_tmu_clock_event_suspend;
+        ced->resume = sh_tmu_clock_event_resume;
 
         dev_info(&p->pdev->dev, "used for clock events\n");
 
@@ -407,8 +449,12 @@ static int __devinit sh_tmu_probe(struct platform_device *pdev)
         struct sh_tmu_priv *p = platform_get_drvdata(pdev);
         int ret;
 
-        if (!is_early_platform_device(pdev))
-                pm_genpd_dev_always_on(&pdev->dev, true);
+        if (!is_early_platform_device(pdev)) {
+                struct sh_timer_config *cfg = pdev->dev.platform_data;
+
+                if (cfg->clocksource_rating || cfg->clockevent_rating)
+                        pm_genpd_dev_always_on(&pdev->dev, true);
+        }
 
         if (p) {
                 dev_info(&pdev->dev, "kept as earlytimer\n");