*/
 
 #include <linux/mfd/ingenic-tcu.h>
+#include <linux/mfd/syscon.h>
 #include <linux/module.h>
 #include <linux/moduleparam.h>
 #include <linux/types.h>
 #include <linux/slab.h>
 #include <linux/err.h>
 #include <linux/of.h>
+#include <linux/regmap.h>
 
 #define DEFAULT_HEARTBEAT 5
 #define MAX_HEARTBEAT     2048
 
 struct jz4740_wdt_drvdata {
        struct watchdog_device wdt;
-       void __iomem *base;
+       struct regmap *map;
        struct clk *clk;
        unsigned long clk_rate;
 };
 {
        struct jz4740_wdt_drvdata *drvdata = watchdog_get_drvdata(wdt_dev);
 
-       writew(0x0, drvdata->base + TCU_REG_WDT_TCNT);
+       regmap_write(drvdata->map, TCU_REG_WDT_TCNT, 0);
+
        return 0;
 }
 
 {
        struct jz4740_wdt_drvdata *drvdata = watchdog_get_drvdata(wdt_dev);
        u16 timeout_value = (u16)(drvdata->clk_rate * new_timeout);
-       u8 tcer;
+       unsigned int tcer;
 
-       tcer = readb(drvdata->base + TCU_REG_WDT_TCER);
-       writeb(0x0, drvdata->base + TCU_REG_WDT_TCER);
+       regmap_read(drvdata->map, TCU_REG_WDT_TCER, &tcer);
+       regmap_write(drvdata->map, TCU_REG_WDT_TCER, 0);
 
-       writew((u16)timeout_value, drvdata->base + TCU_REG_WDT_TDR);
-       writew(0x0, drvdata->base + TCU_REG_WDT_TCNT);
+       regmap_write(drvdata->map, TCU_REG_WDT_TDR, timeout_value);
+       regmap_write(drvdata->map, TCU_REG_WDT_TCNT, 0);
 
        if (tcer & TCU_WDT_TCER_TCEN)
-               writeb(TCU_WDT_TCER_TCEN, drvdata->base + TCU_REG_WDT_TCER);
+               regmap_write(drvdata->map, TCU_REG_WDT_TCER, TCU_WDT_TCER_TCEN);
 
        wdt_dev->timeout = new_timeout;
        return 0;
 static int jz4740_wdt_start(struct watchdog_device *wdt_dev)
 {
        struct jz4740_wdt_drvdata *drvdata = watchdog_get_drvdata(wdt_dev);
+       unsigned int tcer;
        int ret;
-       u8 tcer;
 
        ret = clk_prepare_enable(drvdata->clk);
        if (ret)
                return ret;
 
-       tcer = readb(drvdata->base + TCU_REG_WDT_TCER);
+       regmap_read(drvdata->map, TCU_REG_WDT_TCER, &tcer);
 
        jz4740_wdt_set_timeout(wdt_dev, wdt_dev->timeout);
 
        /* Start watchdog if it wasn't started already */
        if (!(tcer & TCU_WDT_TCER_TCEN))
-               writeb(TCU_WDT_TCER_TCEN, drvdata->base + TCU_REG_WDT_TCER);
+               regmap_write(drvdata->map, TCU_REG_WDT_TCER, TCU_WDT_TCER_TCEN);
 
        return 0;
 }
 {
        struct jz4740_wdt_drvdata *drvdata = watchdog_get_drvdata(wdt_dev);
 
-       writeb(0x0, drvdata->base + TCU_REG_WDT_TCER);
+       regmap_write(drvdata->map, TCU_REG_WDT_TCER, 0);
        clk_disable_unprepare(drvdata->clk);
 
        return 0;
        watchdog_set_nowayout(jz4740_wdt, nowayout);
        watchdog_set_drvdata(jz4740_wdt, drvdata);
 
-       drvdata->base = devm_platform_ioremap_resource(pdev, 0);
-       if (IS_ERR(drvdata->base))
-               return PTR_ERR(drvdata->base);
+       drvdata->map = device_node_to_regmap(dev->parent->of_node);
+       if (!drvdata->map) {
+               dev_err(dev, "regmap not found\n");
+               return -EINVAL;
+       }
 
        return devm_watchdog_register_device(dev, &drvdata->wdt);
 }