arm: stm32mp: stm32prog: update multiplier is part-size is above SZ_1G

Set multiplier to 'G' if part->size if above SZ_1G.

Signed-off-by: Patrice Chotard <patrice.chotard@foss.st.com>
Reviewed-by: Patrick Delaunay <patrick.delaunay@foss.st.com>

diff --git a/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c b/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c
index f0e019e..353aecc 100644 (file)
--- a/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c
+++ b/arch/arm/mach-stm32mp/cmd_stm32prog/stm32prog.c
@@ -1229,7 +1229,10 @@ static int stm32prog_alt_add(struct stm32prog_data *data,
        char multiplier,  type;
 
        /* max 3 digit for sector size */
-       if (part->size > SZ_1M) {
+       if (part->size > SZ_1G) {
+               size = (u32)(part->size / SZ_1G);
+               multiplier = 'G';
+       } else if (part->size > SZ_1M) {
                size = (u32)(part->size / SZ_1M);
                multiplier = 'M';
        } else if (part->size > SZ_1K) {