#include <linux/types.h>
 #include <linux/bitops.h>
 #include <linux/interrupt.h>
-#include <linux/gpio.h>
+#include <linux/gpio/driver.h>
 #include <linux/acpi.h>
 #include <linux/platform_device.h>
 #include <linux/seq_file.h>
        struct byt_gpio_pin_context     *saved_context;
 };
 
-#define to_byt_gpio(c) container_of(c, struct byt_gpio, chip)
-
 static void __iomem *byt_gpio_reg(struct gpio_chip *chip, unsigned offset,
                                 int reg)
 {
-       struct byt_gpio *vg = to_byt_gpio(chip);
+       struct byt_gpio *vg = gpiochip_get_data(chip);
        u32 reg_offset;
 
        if (reg == BYT_INT_STAT_REG)
 
 static int byt_gpio_request(struct gpio_chip *chip, unsigned offset)
 {
-       struct byt_gpio *vg = to_byt_gpio(chip);
+       struct byt_gpio *vg = gpiochip_get_data(chip);
        void __iomem *reg = byt_gpio_reg(chip, offset, BYT_CONF0_REG);
        u32 value, gpio_mux;
        unsigned long flags;
 
 static void byt_gpio_free(struct gpio_chip *chip, unsigned offset)
 {
-       struct byt_gpio *vg = to_byt_gpio(chip);
+       struct byt_gpio *vg = gpiochip_get_data(chip);
 
        byt_gpio_clear_triggering(vg, offset);
        pm_runtime_put(&vg->pdev->dev);
 
 static int byt_irq_type(struct irq_data *d, unsigned type)
 {
-       struct byt_gpio *vg = to_byt_gpio(irq_data_get_irq_chip_data(d));
+       struct byt_gpio *vg = gpiochip_get_data(irq_data_get_irq_chip_data(d));
        u32 offset = irqd_to_hwirq(d);
        u32 value;
        unsigned long flags;
 static int byt_gpio_get(struct gpio_chip *chip, unsigned offset)
 {
        void __iomem *reg = byt_gpio_reg(chip, offset, BYT_VAL_REG);
-       struct byt_gpio *vg = to_byt_gpio(chip);
+       struct byt_gpio *vg = gpiochip_get_data(chip);
        unsigned long flags;
        u32 val;
 
 
 static void byt_gpio_set(struct gpio_chip *chip, unsigned offset, int value)
 {
-       struct byt_gpio *vg = to_byt_gpio(chip);
+       struct byt_gpio *vg = gpiochip_get_data(chip);
        void __iomem *reg = byt_gpio_reg(chip, offset, BYT_VAL_REG);
        unsigned long flags;
        u32 old_val;
 
 static int byt_gpio_direction_input(struct gpio_chip *chip, unsigned offset)
 {
-       struct byt_gpio *vg = to_byt_gpio(chip);
+       struct byt_gpio *vg = gpiochip_get_data(chip);
        void __iomem *reg = byt_gpio_reg(chip, offset, BYT_VAL_REG);
        unsigned long flags;
        u32 value;
 static int byt_gpio_direction_output(struct gpio_chip *chip,
                                     unsigned gpio, int value)
 {
-       struct byt_gpio *vg = to_byt_gpio(chip);
+       struct byt_gpio *vg = gpiochip_get_data(chip);
        void __iomem *conf_reg = byt_gpio_reg(chip, gpio, BYT_CONF0_REG);
        void __iomem *reg = byt_gpio_reg(chip, gpio, BYT_VAL_REG);
        unsigned long flags;
 
 static void byt_gpio_dbg_show(struct seq_file *s, struct gpio_chip *chip)
 {
-       struct byt_gpio *vg = to_byt_gpio(chip);
+       struct byt_gpio *vg = gpiochip_get_data(chip);
        int i;
        u32 conf0, val, offs;
 
 static void byt_gpio_irq_handler(struct irq_desc *desc)
 {
        struct irq_data *data = irq_desc_get_irq_data(desc);
-       struct byt_gpio *vg = to_byt_gpio(irq_desc_get_handler_data(desc));
+       struct byt_gpio *vg = gpiochip_get_data(irq_desc_get_handler_data(desc));
        struct irq_chip *chip = irq_data_get_irq_chip(data);
        u32 base, pin;
        void __iomem *reg;
 static void byt_irq_ack(struct irq_data *d)
 {
        struct gpio_chip *gc = irq_data_get_irq_chip_data(d);
-       struct byt_gpio *vg = to_byt_gpio(gc);
+       struct byt_gpio *vg = gpiochip_get_data(gc);
        unsigned offset = irqd_to_hwirq(d);
        void __iomem *reg;
 
 static void byt_irq_unmask(struct irq_data *d)
 {
        struct gpio_chip *gc = irq_data_get_irq_chip_data(d);
-       struct byt_gpio *vg = to_byt_gpio(gc);
+       struct byt_gpio *vg = gpiochip_get_data(gc);
        unsigned offset = irqd_to_hwirq(d);
        unsigned long flags;
        void __iomem *reg;
 static void byt_irq_mask(struct irq_data *d)
 {
        struct gpio_chip *gc = irq_data_get_irq_chip_data(d);
-       struct byt_gpio *vg = to_byt_gpio(gc);
+       struct byt_gpio *vg = gpiochip_get_data(gc);
 
        byt_gpio_clear_triggering(vg, irqd_to_hwirq(d));
 }
                                       sizeof(*vg->saved_context), GFP_KERNEL);
 #endif
 
-       ret = gpiochip_add(gc);
+       ret = gpiochip_add_data(gc, vg);
        if (ret) {
                dev_err(&pdev->dev, "failed adding byt-gpio chip\n");
                return ret;