[PATCH] rockchip: sdram: fix LPDDR5 bank info for sys_reg version 3

[Kever Yang]

From: YouMin Chen <cym at rock-chips.com>

This patch add support for additional bank info used by LPDDR5.

Signed-off-by: YouMin Chen <cym at rock-chips.com>
Signed-off-by: Kever Yang <kever.yang at rock-chips.com>
---

 arch/arm/mach-rockchip/sdram.c | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/arch/arm/mach-rockchip/sdram.c b/arch/arm/mach-rockchip/sdram.c
index 99ecbdc3412..d65c48bf515 100644
--- a/arch/arm/mach-rockchip/sdram.c
+++ b/arch/arm/mach-rockchip/sdram.c
@@ -110,6 +110,13 @@ size_t rockchip_sdram_size(phys_addr_t reg)
 			  SYS_REG_COL_MASK);
 		cs1_col = cs0_col;
 		bk = 3 - ((sys_reg2 >> SYS_REG_BK_SHIFT(ch)) & SYS_REG_BK_MASK);
+		/*
+		 * SYS_REG_BK(Version 3):
+		 * 1) Except LPDDR5 0:8bank(bk=3), 1:4bank(bk=2)
+		 * 2) LPDDR5 0:8bank(bk=3), 1:16bank(bk=4)
+		 */
+		if (version == 3 && dram_type == LPDDR5 && bk == 2)
+			bk = 4;
 		if (version >= 2) {
 			cs1_col = 9 + (sys_reg3 >> SYS_REG_CS1_COL_SHIFT(ch) &
 				  SYS_REG_CS1_COL_MASK);
-- 
2.25.1