/* Forward declare this for the dma callbacks*/
 static void lpuart_dma_tx_complete(void *arg);
 
-static inline u32 lpuart32_read(struct uart_port *port, u32 reg_off)
-{
-       return ioread32be(port->membase + reg_off);
+static inline u32 lpuart32_read(struct uart_port *port, u32 off)
+{
+       switch (port->iotype) {
+       case UPIO_MEM32:
+               return readl(port->membase + off);
+       case UPIO_MEM32BE:
+               return ioread32be(port->membase + off);
+       default:
+               return 0;
+       }
 }
 
 static inline void lpuart32_write(struct uart_port *port, u32 val,
-                                 u32 reg_off)
-{
-       iowrite32be(val, port->membase + reg_off);
+                                 u32 off)
+{
+       switch (port->iotype) {
+       case UPIO_MEM32:
+               writel(val, port->membase + off);
+               break;
+       case UPIO_MEM32BE:
+               iowrite32be(val, port->membase + off);
+               break;
+       }
 }
 
 static void lpuart_stop_tx(struct uart_port *port)
 
        spin_lock_irqsave(&sport->port.lock, flags);
        if (sport->port.x_char) {
-               if (sport->port.iotype & UPIO_MEM32BE)
+               if (sport->port.iotype & (UPIO_MEM32 | UPIO_MEM32BE))
                        lpuart32_write(&sport->port, sport->port.x_char, UARTDATA);
                else
                        writeb(sport->port.x_char, sport->port.membase + UARTDR);
        }
 
        if (uart_circ_empty(xmit) || uart_tx_stopped(&sport->port)) {
-               if (sport->port.iotype & UPIO_MEM32BE)
+               if (sport->port.iotype & (UPIO_MEM32 | UPIO_MEM32BE))
                        lpuart32_stop_tx(&sport->port);
                else
                        lpuart_stop_tx(&sport->port);
                goto out;
        }
 
-       if (sport->port.iotype & UPIO_MEM32BE)
+       if (sport->port.iotype & (UPIO_MEM32 | UPIO_MEM32BE))
                lpuart32_transmit_buffer(sport);
        else
                lpuart_transmit_buffer(sport);
        if (options)
                uart_parse_options(options, &baud, &parity, &bits, &flow);
        else
-               if (sport->port.iotype & UPIO_MEM32BE)
+               if (sport->port.iotype & (UPIO_MEM32 | UPIO_MEM32BE))
                        lpuart32_console_get_options(sport, &baud, &parity, &bits);
                else
                        lpuart_console_get_options(sport, &baud, &parity, &bits);
 
-       if (sport->port.iotype & UPIO_MEM32BE)
+       if (sport->port.iotype & (UPIO_MEM32 | UPIO_MEM32BE))
                lpuart32_setup_watermark(sport);
        else
                lpuart_setup_watermark(sport);
        if (!device->port.membase)
                return -ENODEV;
 
+       device->port.iotype = UPIO_MEM32BE;
        device->con->write = lpuart32_early_write;
        return 0;
 }
        }
        sport->port.irq = ret;
        sport->port.iotype = sdata->iotype;
-       if (sport->port.iotype & UPIO_MEM32BE)
+       if (sport->port.iotype & (UPIO_MEM32 | UPIO_MEM32BE))
                sport->port.ops = &lpuart32_pops;
        else
                sport->port.ops = &lpuart_pops;
 
        platform_set_drvdata(pdev, &sport->port);
 
-       if (sport->port.iotype & UPIO_MEM32BE)
+       if (sport->port.iotype & (UPIO_MEM32 | UPIO_MEM32BE))
                lpuart_reg.cons = LPUART32_CONSOLE;
        else
                lpuart_reg.cons = LPUART_CONSOLE;
        struct lpuart_port *sport = dev_get_drvdata(dev);
        unsigned long temp;
 
-       if (sport->port.iotype & UPIO_MEM32BE) {
+       if (sport->port.iotype & (UPIO_MEM32 | UPIO_MEM32BE)) {
                /* disable Rx/Tx and interrupts */
                temp = lpuart32_read(&sport->port, UARTCTRL);
                temp &= ~(UARTCTRL_TE | UARTCTRL_TIE | UARTCTRL_TCIE);
        if (sport->port.suspended && !sport->port.irq_wake)
                clk_prepare_enable(sport->clk);
 
-       if (sport->port.iotype & UPIO_MEM32BE) {
+       if (sport->port.iotype & (UPIO_MEM32 | UPIO_MEM32BE)) {
                lpuart32_setup_watermark(sport);
                temp = lpuart32_read(&sport->port, UARTCTRL);
                temp |= (UARTCTRL_RIE | UARTCTRL_TIE | UARTCTRL_RE |