2 files changed, 22 insertions, 1 deletions

diff --git a/arch/m68knommu/platform/5307/head.S b/arch/m68knommu/platform/5307/head.S
index f2edb6498cd9..b9aa0ca29bfb 100644
--- a/arch/m68knommu/platform/5307/head.S
+++ b/arch/m68knommu/platform/5307/head.S
@@ -64,6 +64,26 @@
         negl    %d0                     /* negate bits */
 .endm
 
+#elif defined(CONFIG_M520x)
+.macro GET_MEM_SIZE
+        clrl    %d0
+        movel   MCF_MBAR+MCFSIM_SDCS0, %d2 /* Get SDRAM chip select 0 config */
+        andl    #0x1f, %d2              /* Get only the chip select size */
+        beq     3f                      /* Check if it is enabled */
+        addql   #1, %d2                 /* Form exponent */
+        moveql  #1, %d0
+        lsll    %d2, %d0                /* 2 ^ exponent */
+3:
+        movel   MCF_MBAR+MCFSIM_SDCS1, %d2 /* Get SDRAM chip select 1 config */
+        andl    #0x1f, %d2              /* Get only the chip select size */
+        beq     4f                      /* Check if it is enabled */
+        addql   #1, %d2                 /* Form exponent */
+        moveql  #1, %d1
+        lsll    %d2, %d1                /* 2 ^ exponent */
+        addl    %d1, %d0                /* Total size of SDRAM in d0 */
+4:
+.endm
+
 #else
 #error "ERROR: I don't know how to probe your boards memory size?"
 #endif
diff --git a/arch/m68knommu/platform/68360/head-ram.S b/arch/m68knommu/platform/68360/head-ram.S
index 2ea51479f13a..2ef06242398b 100644
--- a/arch/m68knommu/platform/68360/head-ram.S
+++ b/arch/m68knommu/platform/68360/head-ram.S
@@ -25,6 +25,7 @@
 .global _periph_base
 
 #define RAMEND                      (CONFIG_RAMBASE + CONFIG_RAMSIZE)
+#define ROMEND                      (CONFIG_ROMBASE + CONFIG_ROMSIZE)
 
 #define REGB                        0x1000
 #define PEPAR                       (_dprbase + REGB + 0x0016)
@@ -175,7 +176,7 @@ configure_chip_select_0:
         move.l  %d0, BR0
 
 configure_chip_select_1:
-        move.l  #__rom_end, %d0
+        move.l  #ROMEND, %d0
         subi.l  #__rom_start, %d0
         subq.l  #0x01, %d0
         eori.l  #SIM_OR_MASK, %d0