--- zzzz-none-000/linux-4.4.60/arch/arm/kernel/process.c	2017-04-08 07:53:53.000000000 +0000
+++ scorpion-7490-727/linux-4.4.60/arch/arm/kernel/process.c	2021-02-04 17:41:59.000000000 +0000
@@ -37,6 +37,17 @@
 #include <asm/tls.h>
 #include <asm/vdso.h>
 
+#include <linux/threads.h>
+#include <linux/smp.h>
+#include <asm/mach_avm.h>
+
+#include <avm/profile/profile.h>
+#include <avm/power/power.h>
+
+#if defined(CONFIG_AVM_ENHANCED)
+#include <asm/avm_enh/avm_enh.h>
+#endif/*--- #if defined(CONFIG_AVM_ENHANCED) ---*/
+
 #ifdef CONFIG_CC_STACKPROTECTOR
 #include <linux/stackprotector.h>
 unsigned long __stack_chk_guard __read_mostly;
@@ -66,10 +77,14 @@
 
 void arch_cpu_idle(void)
 {
+	avm_cpu_wait_start();
+
 	if (arm_pm_idle)
-		arm_pm_idle();
+	arm_pm_idle();
 	else
-		cpu_do_idle();
+	cpu_do_idle();
+
+	avm_cpu_wait_end();
 	local_irq_enable();
 }
 
@@ -178,6 +193,9 @@
 		printk("Control: %08x%s\n", ctrl, buf);
 	}
 #endif
+#if defined(CONFIG_AVM_ENHANCED)
+	show_register_memoryclassifier(regs);
+#endif/*--- #if defined(CONFIG_AVM_ENHANCED) ---*/
 }
 
 void show_regs(struct pt_regs * regs)