#include <linux/io.h>
 #include <linux/slab.h>
 #include <linux/err.h>
-#include <linux/gpio.h>
+#include <linux/gpio/driver.h>
 #include <linux/device.h>
 #include <linux/pinctrl/pinctrl.h>
 #include <linux/pinctrl/pinmux.h>
 /* gpiolib gpio_set callback function */
 static void exynos5440_gpio_set(struct gpio_chip *gc, unsigned offset, int value)
 {
-       struct exynos5440_pinctrl_priv_data *priv = dev_get_drvdata(gc->parent);
+       struct exynos5440_pinctrl_priv_data *priv = gpiochip_get_data(gc);
        void __iomem *base = priv->reg_base;
        u32 data;
 
 /* gpiolib gpio_get callback function */
 static int exynos5440_gpio_get(struct gpio_chip *gc, unsigned offset)
 {
-       struct exynos5440_pinctrl_priv_data *priv = dev_get_drvdata(gc->parent);
+       struct exynos5440_pinctrl_priv_data *priv = gpiochip_get_data(gc);
        void __iomem *base = priv->reg_base;
        u32 data;
 
 /* gpiolib gpio_direction_input callback function */
 static int exynos5440_gpio_direction_input(struct gpio_chip *gc, unsigned offset)
 {
-       struct exynos5440_pinctrl_priv_data *priv = dev_get_drvdata(gc->parent);
+       struct exynos5440_pinctrl_priv_data *priv = gpiochip_get_data(gc);
        void __iomem *base = priv->reg_base;
        u32 data;
 
 static int exynos5440_gpio_direction_output(struct gpio_chip *gc, unsigned offset,
                                                        int value)
 {
-       struct exynos5440_pinctrl_priv_data *priv = dev_get_drvdata(gc->parent);
+       struct exynos5440_pinctrl_priv_data *priv = gpiochip_get_data(gc);
        void __iomem *base = priv->reg_base;
        u32 data;
 
 /* gpiolib gpio_to_irq callback function */
 static int exynos5440_gpio_to_irq(struct gpio_chip *gc, unsigned offset)
 {
-       struct exynos5440_pinctrl_priv_data *priv = dev_get_drvdata(gc->parent);
+       struct exynos5440_pinctrl_priv_data *priv = gpiochip_get_data(gc);
        unsigned int virq;
 
        if (offset < 16 || offset > 23)
        gc->to_irq = exynos5440_gpio_to_irq;
        gc->label = "gpiolib-exynos5440";
        gc->owner = THIS_MODULE;
-       ret = gpiochip_add(gc);
+       ret = gpiochip_add_data(gc, priv);
        if (ret) {
                dev_err(&pdev->dev, "failed to register gpio_chip %s, error "
                                        "code: %d\n", gc->label, ret);