#include <linux/gpio/driver.h>
 #include <linux/interrupt.h>
 #include <linux/io.h>
+#include <linux/mfd/ocelot.h>
 #include <linux/of_device.h>
 #include <linux/of_irq.h>
 #include <linux/of_platform.h>
        struct ocelot_pinctrl *info;
        struct reset_control *reset;
        struct regmap *pincfg;
-       void __iomem *base;
        int ret;
        struct regmap_config regmap_config = {
                .reg_bits = 32,
                                     "Failed to get reset\n");
        reset_control_reset(reset);
 
-       base = devm_ioremap_resource(dev,
-                       platform_get_resource(pdev, IORESOURCE_MEM, 0));
-       if (IS_ERR(base))
-               return PTR_ERR(base);
-
        info->stride = 1 + (info->desc->npins - 1) / 32;
 
        regmap_config.max_register = OCELOT_GPIO_SD_MAP * info->stride + 15 * 4;
 
-       info->map = devm_regmap_init_mmio(dev, base, ®map_config);
-       if (IS_ERR(info->map)) {
-               dev_err(dev, "Failed to create regmap\n");
-               return PTR_ERR(info->map);
-       }
+       info->map = ocelot_regmap_from_resource(pdev, 0, ®map_config);
+       if (IS_ERR(info->map))
+               return dev_err_probe(dev, PTR_ERR(info->map),
+                                    "Failed to create regmap\n");
        dev_set_drvdata(dev, info->map);
        info->dev = dev;