*/
 
 #include <linux/err.h>
-#include <linux/gpio.h>
+#include <linux/gpio/driver.h>
 #include <linux/interrupt.h>
 #include <linux/io.h>
 #include <linux/irq.h>
 
 static int wmt_gpio_get_direction(struct gpio_chip *chip, unsigned offset)
 {
-       struct wmt_pinctrl_data *data = dev_get_drvdata(chip->parent);
+       struct wmt_pinctrl_data *data = gpiochip_get_data(chip);
        u32 bank = WMT_BANK_FROM_PIN(offset);
        u32 bit = WMT_BIT_FROM_PIN(offset);
        u32 reg_dir = data->banks[bank].reg_dir;
 
 static int wmt_gpio_get_value(struct gpio_chip *chip, unsigned offset)
 {
-       struct wmt_pinctrl_data *data = dev_get_drvdata(chip->parent);
+       struct wmt_pinctrl_data *data = gpiochip_get_data(chip);
        u32 bank = WMT_BANK_FROM_PIN(offset);
        u32 bit = WMT_BIT_FROM_PIN(offset);
        u32 reg_data_in = data->banks[bank].reg_data_in;
 static void wmt_gpio_set_value(struct gpio_chip *chip, unsigned offset,
                               int val)
 {
-       struct wmt_pinctrl_data *data = dev_get_drvdata(chip->parent);
+       struct wmt_pinctrl_data *data = gpiochip_get_data(chip);
        u32 bank = WMT_BANK_FROM_PIN(offset);
        u32 bit = WMT_BIT_FROM_PIN(offset);
        u32 reg_data_out = data->banks[bank].reg_data_out;
                return PTR_ERR(data->pctl_dev);
        }
 
-       err = gpiochip_add(&data->gpio_chip);
+       err = gpiochip_add_data(&data->gpio_chip, data);
        if (err) {
                dev_err(&pdev->dev, "could not add GPIO chip\n");
                goto fail_gpio;