#include <linux/delay.h>
 #include <linux/pm.h>
 #include <linux/gcd.h>
+#include <linux/gpio/driver.h>
 #include <linux/gpio.h>
 #include <linux/i2c.h>
 #include <linux/regmap.h>
 }
 
 #ifdef CONFIG_GPIOLIB
-static inline struct wm8996_priv *gpio_to_wm8996(struct gpio_chip *chip)
-{
-       return container_of(chip, struct wm8996_priv, gpio_chip);
-}
-
 static void wm8996_gpio_set(struct gpio_chip *chip, unsigned offset, int value)
 {
-       struct wm8996_priv *wm8996 = gpio_to_wm8996(chip);
+       struct wm8996_priv *wm8996 = gpiochip_get_data(chip);
 
        regmap_update_bits(wm8996->regmap, WM8996_GPIO_1 + offset,
                           WM8996_GP1_LVL, !!value << WM8996_GP1_LVL_SHIFT);
 static int wm8996_gpio_direction_out(struct gpio_chip *chip,
                                     unsigned offset, int value)
 {
-       struct wm8996_priv *wm8996 = gpio_to_wm8996(chip);
+       struct wm8996_priv *wm8996 = gpiochip_get_data(chip);
        int val;
 
        val = (1 << WM8996_GP1_FN_SHIFT) | (!!value << WM8996_GP1_LVL_SHIFT);
 
 static int wm8996_gpio_get(struct gpio_chip *chip, unsigned offset)
 {
-       struct wm8996_priv *wm8996 = gpio_to_wm8996(chip);
+       struct wm8996_priv *wm8996 = gpiochip_get_data(chip);
        unsigned int reg;
        int ret;
 
 
 static int wm8996_gpio_direction_in(struct gpio_chip *chip, unsigned offset)
 {
-       struct wm8996_priv *wm8996 = gpio_to_wm8996(chip);
+       struct wm8996_priv *wm8996 = gpiochip_get_data(chip);
 
        return regmap_update_bits(wm8996->regmap, WM8996_GPIO_1 + offset,
                                  WM8996_GP1_FN_MASK | WM8996_GP1_DIR,
        else
                wm8996->gpio_chip.base = -1;
 
-       ret = gpiochip_add(&wm8996->gpio_chip);
+       ret = gpiochip_add_data(&wm8996->gpio_chip, wm8996);
        if (ret != 0)
                dev_err(wm8996->dev, "Failed to add GPIOs: %d\n", ret);
 }