#include <linux/io.h>
 #include <linux/of.h>
 #include <linux/of_gpio.h>
+#include <linux/gpio/driver.h>
+/* FIXME: needed for gpio_to_chip() get rid of this */
 #include <linux/gpio.h>
 #include <linux/slab.h>
 #include <linux/export.h>
        struct qe_pio_regs saved_regs;
 };
 
-static inline struct qe_gpio_chip *
-to_qe_gpio_chip(struct of_mm_gpio_chip *mm_gc)
-{
-       return container_of(mm_gc, struct qe_gpio_chip, mm_gc);
-}
-
 static void qe_gpio_save_regs(struct of_mm_gpio_chip *mm_gc)
 {
-       struct qe_gpio_chip *qe_gc = to_qe_gpio_chip(mm_gc);
+       struct qe_gpio_chip *qe_gc = gpiochip_get_data(&mm_gc->gc);
        struct qe_pio_regs __iomem *regs = mm_gc->regs;
 
        qe_gc->cpdata = in_be32(®s->cpdata);
 static void qe_gpio_set(struct gpio_chip *gc, unsigned int gpio, int val)
 {
        struct of_mm_gpio_chip *mm_gc = to_of_mm_gpio_chip(gc);
-       struct qe_gpio_chip *qe_gc = to_qe_gpio_chip(mm_gc);
+       struct qe_gpio_chip *qe_gc = gpiochip_get_data(gc);
        struct qe_pio_regs __iomem *regs = mm_gc->regs;
        unsigned long flags;
        u32 pin_mask = 1 << (QE_PIO_PINS - 1 - gpio);
 static int qe_gpio_dir_in(struct gpio_chip *gc, unsigned int gpio)
 {
        struct of_mm_gpio_chip *mm_gc = to_of_mm_gpio_chip(gc);
-       struct qe_gpio_chip *qe_gc = to_qe_gpio_chip(mm_gc);
+       struct qe_gpio_chip *qe_gc = gpiochip_get_data(gc);
        unsigned long flags;
 
        spin_lock_irqsave(&qe_gc->lock, flags);
 static int qe_gpio_dir_out(struct gpio_chip *gc, unsigned int gpio, int val)
 {
        struct of_mm_gpio_chip *mm_gc = to_of_mm_gpio_chip(gc);
-       struct qe_gpio_chip *qe_gc = to_qe_gpio_chip(mm_gc);
+       struct qe_gpio_chip *qe_gc = gpiochip_get_data(gc);
        unsigned long flags;
 
        qe_gpio_set(gc, gpio, val);
        }
 
        mm_gc = to_of_mm_gpio_chip(gc);
-       qe_gc = to_qe_gpio_chip(mm_gc);
+       qe_gc = gpiochip_get_data(gc);
 
        spin_lock_irqsave(&qe_gc->lock, flags);
 
                gc->get = qe_gpio_get;
                gc->set = qe_gpio_set;
 
-               ret = of_mm_gpiochip_add(np, mm_gc);
+               ret = of_mm_gpiochip_add_data(np, mm_gc, qe_gc);
                if (ret)
                        goto err;
                continue;