arm: mach-k3: j721e: Fix unlocking control module registers

In main control mmr there is no partition 4 and partition 6 is available
only on J721e. Fix the same in ctrl_mmr_unlock function

Signed-off-by: Lokesh Vutla <lokeshvutla@ti.com>
Reviewed-by: Suman Anna <s-anna@ti.com>

diff --git a/arch/arm/mach-k3/j721e_init.c b/arch/arm/mach-k3/j721e_init.c
index 3b15da2d7cdf68fd67fb16a63dfb7765d8f8ead4..63a31c18ce22f3180c55abf6b04bda10b120fb5e 100644 (file)
--- a/arch/arm/mach-k3/j721e_init.c
+++ b/arch/arm/mach-k3/j721e_init.c
@@ -87,9 +87,9 @@ static void ctrl_mmr_unlock(void)
        mmr_unlock(CTRL_MMR0_BASE, 1);
        mmr_unlock(CTRL_MMR0_BASE, 2);
        mmr_unlock(CTRL_MMR0_BASE, 3);
-       mmr_unlock(CTRL_MMR0_BASE, 4);
        mmr_unlock(CTRL_MMR0_BASE, 5);
-       mmr_unlock(CTRL_MMR0_BASE, 6);
+       if (soc_is_j721e())
+               mmr_unlock(CTRL_MMR0_BASE, 6);
        mmr_unlock(CTRL_MMR0_BASE, 7);
 }