#include <linux/module.h>
 #include <linux/pm.h>
 #include <linux/property.h>
+#include <linux/regulator/consumer.h>
 #include <linux/slab.h>
 #include <linux/spinlock.h>
 #include <dt-bindings/mux/mux.h>
        struct irq_domain *irq;
        unsigned int irq_mask;
        raw_spinlock_t lock;
+       struct regulator *supply;
 };
 
 /* Provide specs for the MAX735x, PCA954x and PCA984x types we know about */
        struct pca954x *data = i2c_mux_priv(muxc);
        int c, irq;
 
+       regulator_disable(data->supply);
+
        if (data->irq) {
                for (c = 0; c < data->chip->nchans; c++) {
                        irq = irq_find_mapping(data->irq, c);
        i2c_set_clientdata(client, muxc);
        data->client = client;
 
+       data->supply = devm_regulator_get(dev, "vdd");
+       if (IS_ERR(data->supply))
+               return dev_err_probe(dev, PTR_ERR(data->supply),
+                               "Failed to request regulator\n");
+
+       ret = regulator_enable(data->supply);
+       if (ret)
+               return dev_err_probe(dev, ret,
+                                    "Failed to enable vdd supply\n");
+
        /* Reset the mux if a reset GPIO is specified. */
        gpio = devm_gpiod_get_optional(dev, "reset", GPIOD_OUT_HIGH);
-       if (IS_ERR(gpio))
-               return PTR_ERR(gpio);
+       if (IS_ERR(gpio)) {
+               ret = PTR_ERR(gpio);
+               goto fail_cleanup;
+       }
        if (gpio) {
                udelay(1);
                gpiod_set_value_cansleep(gpio, 0);
 
                ret = i2c_get_device_id(client, &id);
                if (ret && ret != -EOPNOTSUPP)
-                       return ret;
+                       goto fail_cleanup;
 
                if (!ret &&
                    (id.manufacturer_id != data->chip->id.manufacturer_id ||
                        dev_warn(dev, "unexpected device id %03x-%03x-%x\n",
                                 id.manufacturer_id, id.part_id,
                                 id.die_revision);
-                       return -ENODEV;
+                       ret = -ENODEV;
+                       goto fail_cleanup;
                }
        }
 
        ret = pca954x_init(client, data);
        if (ret < 0) {
                dev_warn(dev, "probe failed\n");
-               return -ENODEV;
+               ret = -ENODEV;
+               goto fail_cleanup;
        }
 
        ret = pca954x_irq_setup(muxc);