--- zzzz-none-000/linux-5.15.111/drivers/cpuidle/cpuidle.c	2023-05-11 14:00:40.000000000 +0000
+++ puma7-atom-6670-761/linux-5.15.111/drivers/cpuidle/cpuidle.c	2024-02-07 10:22:40.000000000 +0000
@@ -27,6 +27,8 @@
 
 #include "cpuidle.h"
 
+#include <avm/power/power.h>
+
 DEFINE_PER_CPU(struct cpuidle_device *, cpuidle_devices);
 DEFINE_PER_CPU(struct cpuidle_device, cpuidle_dev);
 
@@ -231,6 +233,7 @@
 	trace_cpu_idle(index, dev->cpu);
 	time_start = ns_to_ktime(local_clock());
 
+	avm_cpu_wait_start();
 	stop_critical_timings();
 	if (!(target_state->flags & CPUIDLE_FLAG_RCU_IDLE))
 		rcu_idle_enter();
@@ -238,6 +241,8 @@
 	if (!(target_state->flags & CPUIDLE_FLAG_RCU_IDLE))
 		rcu_idle_exit();
 	start_critical_timings();
+	avm_cpu_wait_end();
+	avm_cpu_wait_info();
 
 	sched_clock_idle_wakeup_event();
 	time_end = ns_to_ktime(local_clock());