marigold: disable deep sleep when the system run at S0 state

Marigold power sequence was missing setting the SLEEP_MASK_AP_RUN flag,
therefore the common sleep mask did not know that the system was in S0
or it would have disabled sleep already.

BRANCH=marigold
BUG=app.clickup.com/t/86ep19bww
TEST=We can't see the HID error message in the Linux dmesg.
TEST=check the sleepmask is 00000001 in S0 state and 00000000 in S5/S3 state

Signed-off-by: Josh-Tsai <josh_tsai@compal.com>

Author: Josh-Tsai

zephyr/program/framework/marigold/src/power_sequence.c

@@ -418,6 +418,12 @@ enum power_state power_handle_state(enum power_state state)
 		/* Call hooks now that rails are up */
 		hook_notify(HOOK_CHIPSET_RESUME);
 
+		/*
+		 * Disable idle task deep sleep. This means that the low
+		 * power idle task will not go into deep sleep while in S0.
+		 */
+		disable_sleep(SLEEP_MASK_AP_RUN);
+
 		/* set the PD chip system power state "S0" */
 		cypd_set_power_active();
 
@@ -465,6 +471,13 @@ enum power_state power_handle_state(enum power_state state)
 		cypd_set_power_active();
 		/* Call hooks now that rails are up */
 		hook_notify(HOOK_CHIPSET_RESUME);
+
+		/*
+		 * Disable idle task deep sleep. This means that the low
+		 * power idle task will not go into deep sleep while in S0.
+		 */
+		disable_sleep(SLEEP_MASK_AP_RUN);
+
 		return POWER_S0;
 
 	case POWER_S0S0ix:
@@ -475,6 +488,13 @@ enum power_state power_handle_state(enum power_state state)
 		cypd_set_power_active();
 		/* Call hooks before we remove power rails */
 		hook_notify(HOOK_CHIPSET_SUSPEND);
+
+		/*
+		 * Enable idle task deep sleep. Allow the low power idle task
+		 * to go into deep sleep in S3 or lower.
+		 */
+		enable_sleep(SLEEP_MASK_AP_RUN);
+
 		return POWER_S0ix;
 #endif
 
@@ -489,6 +509,12 @@ enum power_state power_handle_state(enum power_state state)
 		/* Call hooks before we remove power rails */
 		hook_notify(HOOK_CHIPSET_SUSPEND);
 
+		/*
+		 * Enable idle task deep sleep. Allow the low power idle task
+		 * to go into deep sleep in S3 or lower.
+		 */
+		enable_sleep(SLEEP_MASK_AP_RUN);
+
 		/* set the PD chip system power state "S3" */
 		cypd_set_power_active();
 		return POWER_S3;