arch: arm64: cortex_a: Add CPU load for Cortex-A

- Add calls to sys_trace_idle_exit before leaving idle state
  to track CPU load
- Extend CPU_LOAD to CPU_CORTEX_A in Kconfig

Signed-off-by: Hoang Nguyen <[email protected]>
Signed-off-by: Nhut Nguyen <[email protected]>

Author: Hoang Nguyen

arch/arm64/core/cpu_idle.S

@@ -23,6 +23,11 @@ SECTION_FUNC(TEXT, arch_cpu_idle)
 #endif
 	dsb	sy
 	wfi
+#ifdef CONFIG_TRACING
+	str	lr, [sp, #-16]!
+	bl	sys_trace_idle_exit
+	ldr	lr, [sp], #16
+#endif
 	msr	daifclr, #(DAIFCLR_IRQ_BIT)
 	ret
 #endif
@@ -38,6 +43,11 @@ SECTION_FUNC(TEXT, arch_cpu_atomic_idle)
 	msr	daifset, #(DAIFSET_IRQ_BIT)
 	isb
 	wfe
+#ifdef CONFIG_TRACING
+	str	lr, [sp, #-16]!
+	bl	sys_trace_idle_exit
+	ldr	lr, [sp], #16
+#endif
 	tst	x0, #(DAIF_IRQ_BIT)
 	beq	_irq_disabled
 	msr	daifclr, #(DAIFCLR_IRQ_BIT)

subsys/debug/Kconfig

@@ -353,7 +353,7 @@ config CS_TRACE_DEFMT
 
 config CPU_LOAD
 	select TRACING
-	depends on CPU_CORTEX_M || RISCV
+	depends on CPU_CORTEX_M || RISCV || CPU_CORTEX_A
 	depends on !SMP
 	bool "CPU load measurement"