1 files changed, 6 insertions, 16 deletions

diff --git a/arch/arm/plat-mxc/include/mach/io.h b/arch/arm/plat-mxc/include/mach/io.h
index 4347a87d2bb0..338300b18b00 100644
--- a/arch/arm/plat-mxc/include/mach/io.h
+++ b/arch/arm/plat-mxc/include/mach/io.h
@@ -14,32 +14,22 @@
 /* Allow IO space to be anywhere in the memory */
 #define IO_SPACE_LIMIT 0xffffffff
 
-#if defined(CONFIG_SOC_IMX31) || defined(CONFIG_SOC_IMX35)
-#include <mach/hardware.h>
-
 #define __arch_ioremap __imx_ioremap
 #define __arch_iounmap __iounmap
 
 #define addr_in_module(addr, mod) \
         ((unsigned long)(addr) - mod ## _BASE_ADDR < mod ## _SIZE)
 
+extern void __iomem *(*imx_ioremap)(unsigned long, size_t, unsigned int);
+
 static inline void __iomem *
 __imx_ioremap(unsigned long phys_addr, size_t size, unsigned int mtype)
 {
-        if (mtype == MT_DEVICE && (cpu_is_mx31() || cpu_is_mx35())) {
-                /*
-                 * Access all peripherals below 0x80000000 as nonshared device
-                 * on mx3, but leave l2cc alone.  Otherwise cache corruptions
-                 * can occur.
-                 */
-                if (phys_addr < 0x80000000 &&
-                                !addr_in_module(phys_addr, MX3x_L2CC))
-                        mtype = MT_DEVICE_NONSHARED;
-        }
-
-        return __arm_ioremap(phys_addr, size, mtype);
+        if (imx_ioremap != NULL)
+                return imx_ioremap(phys_addr, size, mtype);
+        else
+                return __arm_ioremap(phys_addr, size, mtype);
 }
-#endif
 
 /* io address mapping macro */
 #define __io(a)         __typesafe_io(a)