sh: Add support showing KByte of flash memory size

Signed-off-by: Nobuhiro Iwamatsu <iwamatsu@nigauri.org>

diff --git a/arch/sh/lib/board.c b/arch/sh/lib/board.c
index 3d201b2e5c03f4b624e5b69c9b4abc585a994d8a..cdac3826c76e20698e49c1dcd041fb0599236b0e 100644 (file)
--- a/arch/sh/lib/board.c
+++ b/arch/sh/lib/board.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2007,2008
+ * Copyright (C) 2007, 2008, 2010
  * Nobuhiro Iwamatsu <iwamatsu@nigauri.org>
  *
  * This program is free software; you can redistribute it and/or
@@ -46,7 +46,11 @@ unsigned long monitor_flash_len = CONFIG_SYS_MONITOR_LEN;
 static int sh_flash_init(void)
 {
        gd->bd->bi_flashsize = flash_init();
-       printf("FLASH: %ldMB\n", gd->bd->bi_flashsize / (1024*1024));
+
+       if (gd->bd->bi_flashsize >= (1024 * 1024))
+               printf("FLASH: %ldMB\n", gd->bd->bi_flashsize / (1024*1024));
+       else
+               printf("FLASH: %ldKB\n", gd->bd->bi_flashsize / 1024);
 
        return 0;
 }