arm64: zynqmp: Fix lockstep mode cpu release functionality

For lockstep mode, cpu_release function is expecting to execute
on R5 core 0, if there is attempt to pass other than R5 core 0,
through an error saying "Lockstep mode should run on R5 core 0 only".

Signed-off-by: Venkatesh Yadav Abbarapu <venkatesh.abbarapu@amd.com>
Link: https://lore.kernel.org/r/20230608032152.980-3-venkatesh.abbarapu@amd.com
Signed-off-by: Michal Simek <michal.simek@amd.com>

diff --git a/arch/arm/mach-zynqmp/mp.c b/arch/arm/mach-zynqmp/mp.c
index 7a12f4b..b06c867 100644 (file)
--- a/arch/arm/mach-zynqmp/mp.c
+++ b/arch/arm/mach-zynqmp/mp.c
@@ -326,6 +326,10 @@ int cpu_release(u32 nr, int argc, char *const argv[])
                flush_dcache_all();
 
                if (!strncmp(argv[1], "lockstep", 8)) {
+                       if (nr != ZYNQMP_CORE_RPU0) {
+                               printf("Lockstep mode should run on ZYNQMP_CORE_RPU0\n");
+                               return 1;
+                       }
                        printf("R5 lockstep mode\n");
                        set_r5_reset(nr, LOCK);
                        set_r5_tcm_mode(LOCK);