sunxi: psci: save context id in cpu_on command

[u-boot] / arch / arm / cpu / armv7 / sunxi / psci.c

diff --git a/arch/arm/cpu/armv7/sunxi/psci.c b/arch/arm/cpu/armv7/sunxi/psci.c
index 97c4b62d472329bce5006e0ac33f07b0f7dd4a03..f3e8f99a7192c92678af8a3ba7446a19836dc461 100644 (file)
--- a/arch/arm/cpu/armv7/sunxi/psci.c
+++ b/arch/arm/cpu/armv7/sunxi/psci.c
@@ -242,14 +242,15 @@ out:
        cp15_write_scr(scr);
 }
 
-int __secure psci_cpu_on(u32 __always_unused unused, u32 mpidr, u32 pc)
+int __secure psci_cpu_on(u32 __always_unused unused, u32 mpidr, u32 pc,
+                        u32 context_id)
 {
        struct sunxi_cpucfg_reg *cpucfg =
                (struct sunxi_cpucfg_reg *)SUNXI_CPUCFG_BASE;
        u32 cpu = (mpidr & 0x3);
 
-       /* store target PC */
-       psci_save_target_pc(cpu, pc);
+       /* store target PC and context id */
+       psci_save(cpu, pc, context_id);
 
        /* Set secondary core power on PC */
        sunxi_set_entry_address(&psci_cpu_entry);