drivers: cpuidle: lpm-levels: Notify sched of low power modes

Call into scheduler APIs when a CPU enters/exits a low power modes to
prevent scheduler from waking up a sleeping CPU

Change-Id: I54af417647c956411125040d0223d0e9293df578
Signed-off-by: Mahesh Sivasubramanian <msivasub@codeaurora.org>
diff --git a/drivers/cpuidle/lpm-levels.c b/drivers/cpuidle/lpm-levels.c
index 4e3ca4f..5d2e918 100644
--- a/drivers/cpuidle/lpm-levels.c
+++ b/drivers/cpuidle/lpm-levels.c
@@ -1346,6 +1346,8 @@
 	struct power_params *pwr_params;
 
 	pwr_params = &cpu->levels[idx].pwr;
+	sched_set_cpu_cstate(dev->cpu, idx + 1,
+			pwr_params->energy_overhead, pwr_params->latency_us);
 
 	cpu_prepare(cpu, idx, true);
 	cluster_prepare(cpu->parent, cpumask, idx, true, start_time);