#include <linux/mutex.h>
 #include <linux/slab.h>
 #include <linux/of.h>
+#include <linux/of_device.h>
 #include <linux/gpio.h>
 #include <linux/pinctrl/machine.h>
 #include <linux/pinctrl/pinconf.h>
 };
 
 static const struct of_device_id sx150x_of_match[] = {
-       { .compatible = "semtech,sx1508q" },
-       { .compatible = "semtech,sx1509q" },
-       { .compatible = "semtech,sx1506q" },
-       { .compatible = "semtech,sx1502q" },
+       { .compatible = "semtech,sx1508q", .data = &sx1508q_device_data },
+       { .compatible = "semtech,sx1509q", .data = &sx1509q_device_data },
+       { .compatible = "semtech,sx1506q", .data = &sx1506q_device_data },
+       { .compatible = "semtech,sx1502q", .data = &sx1502q_device_data },
        {},
 };
 
        struct sx150x_pinctrl *pctl;
        int ret;
 
-       if (!id->driver_data)
-               return -EINVAL;
-
        if (!i2c_check_functionality(client->adapter, i2c_funcs))
                return -ENOSYS;
 
 
        pctl->dev = dev;
        pctl->client = client;
-       pctl->data = (void *)id->driver_data;
+
+       if (dev->of_node)
+               pctl->data = of_device_get_match_data(dev);
+       else
+               pctl->data = (struct sx150x_device_data *)id->driver_data;
+
+       if (!pctl->data)
+               return -EINVAL;
 
        mutex_init(&pctl->lock);