imx: imx8qm/qxp: Recover SPL data section for partition reboot

[oweals/u-boot.git] / arch / arm / mach-imx / imx8 / cpu.c

diff --git a/arch/arm/mach-imx/imx8/cpu.c b/arch/arm/mach-imx/imx8/cpu.c
index 3bd0dee0251f0615e4a64ae6da62922619afffe2..e03193cb4cad3fb77824fca53718c41749e6cb23 100644 (file)
--- a/arch/arm/mach-imx/imx8/cpu.c
+++ b/arch/arm/mach-imx/imx8/cpu.c
@@ -13,6 +13,7 @@
 #include <dm/lists.h>
 #include <dm/uclass.h>
 #include <errno.h>
+#include <spl.h>
 #include <thermal.h>
 #include <asm/arch/sci/sci.h>
 #include <asm/arch/sys_proto.h>
@@ -39,6 +40,10 @@ struct pass_over_info_t *get_pass_over_info(void)
 
 int arch_cpu_init(void)
 {
+#if defined(CONFIG_SPL_BUILD) && defined(CONFIG_SPL_RECOVER_DATA_SECTION)
+       spl_save_restore_data();
+#endif
+
 #ifdef CONFIG_SPL_BUILD
        struct pass_over_info_t *pass_over;