struct sh7372_pm_domain *sh7372_pd = to_sh7372_pd(genpd);
        unsigned int mask = 1 << sh7372_pd->bit_shift;
 
-       if (sh7372_pd->suspend)
-               sh7372_pd->suspend();
+       if (sh7372_pd->suspend) {
+               int ret = sh7372_pd->suspend();
 
-       if (sh7372_pd->stay_on)
-               return 0;
+               if (ret)
+                       return ret;
+       }
 
        if (__raw_readl(PSTR) & mask) {
                unsigned int retry_count;
        unsigned int retry_count;
        int ret = 0;
 
-       if (sh7372_pd->stay_on)
-               goto out;
-
        if (__raw_readl(PSTR) & mask)
                goto out;
 
         return __pd_power_up(to_sh7372_pd(genpd), true);
 }
 
-static void sh7372_a4r_suspend(void)
+static int sh7372_a4r_suspend(void)
 {
        sh7372_intcs_suspend();
        __raw_writel(0x300fffff, WUPRMSK); /* avoid wakeup */
+       return 0;
 }
 
 static bool pd_active_wakeup(struct device *dev)
        .gov = &pm_domain_always_on_gov,
        .suspend = sh7372_a4r_suspend,
        .resume = sh7372_intcs_resume,
-       .stay_on = true,
 };
 
 struct sh7372_pm_domain sh7372_a3rv = {
        .bit_shift = 8,
 };
 
+static int sh7372_a4s_suspend(void)
+{
+       /*
+        * The A4S domain contains the CPU core and therefore it should
+        * only be turned off if the CPU is in use.
+        */
+       return -EBUSY;
+}
+
 struct sh7372_pm_domain sh7372_a4s = {
        .genpd.name = "A4S",
        .bit_shift = 10,
        .gov = &pm_domain_always_on_gov,
        .no_debug = true,
-       .stay_on = true,
+       .suspend = sh7372_a4s_suspend,
 };
 
+static int sh7372_a3sp_suspend(void)
+{
+       /*
+        * Serial consoles make use of SCIF hardware located in A3SP,
+        * keep such power domain on if "no_console_suspend" is set.
+        */
+       return console_suspend_enabled ? -EBUSY : 0;
+}
+
 struct sh7372_pm_domain sh7372_a3sp = {
        .genpd.name = "A3SP",
        .bit_shift = 11,
        .gov = &pm_domain_always_on_gov,
        .no_debug = true,
+       .suspend = sh7372_a3sp_suspend,
 };
 
-static void sh7372_a3sp_init(void)
-{
-       /* serial consoles make use of SCIF hardware located in A3SP,
-        * keep such power domain on if "no_console_suspend" is set.
-        */
-       sh7372_a3sp.stay_on = !console_suspend_enabled;
-}
-
 struct sh7372_pm_domain sh7372_a3sg = {
        .genpd.name = "A3SG",
        .bit_shift = 13,
                /* convert INTC mask and sense to SYSC mask and sense */
                sh7372_setup_sysc(msk, msk2);
 
-               if (!sh7372_a3sp.stay_on &&
+               if (!console_suspend_enabled &&
                    sh7372_a4s.genpd.status == GPD_STATE_POWER_OFF) {
                        /* enter A4S sleep with PLLC0 off */
                        pr_debug("entering A4S\n");
        /* do not convert A3SM, A3SP, A3SG, A4R power down into A4S */
        __raw_writel(0, PDNSEL);
 
-       sh7372_a3sp_init();
-
        sh7372_suspend_init();
        sh7372_cpuidle_init();
 }