#include <linux/module.h>
 #include <linux/of.h>
 #include <linux/of_device.h>
+#include <linux/regulator/coupler.h>
 #include <linux/regulator/driver.h>
 #include <linux/regulator/of_regulator.h>
 #include <linux/sort.h>
 
+#include "internal.h"
+
 struct vctrl_voltage_range {
        int min_uV;
        int max_uV;
 static int vctrl_get_voltage(struct regulator_dev *rdev)
 {
        struct vctrl_data *vctrl = rdev_get_drvdata(rdev);
-       int ctrl_uV = regulator_get_voltage(vctrl->ctrl_reg);
+       int ctrl_uV = regulator_get_voltage_rdev(vctrl->ctrl_reg->rdev);
 
        return vctrl_calc_output_voltage(vctrl, ctrl_uV);
 }
 {
        struct vctrl_data *vctrl = rdev_get_drvdata(rdev);
        struct regulator *ctrl_reg = vctrl->ctrl_reg;
-       int orig_ctrl_uV = regulator_get_voltage(ctrl_reg);
+       int orig_ctrl_uV = regulator_get_voltage_rdev(ctrl_reg->rdev);
        int uV = vctrl_calc_output_voltage(vctrl, orig_ctrl_uV);
        int ret;
 
        if (req_min_uV >= uV || !vctrl->ovp_threshold)
                /* voltage rising or no OVP */
-               return regulator_set_voltage(
-                       ctrl_reg,
+               return regulator_set_voltage_rdev(ctrl_reg->rdev,
                        vctrl_calc_ctrl_voltage(vctrl, req_min_uV),
-                       vctrl_calc_ctrl_voltage(vctrl, req_max_uV));
+                       vctrl_calc_ctrl_voltage(vctrl, req_max_uV),
+                       PM_SUSPEND_ON);
 
        while (uV > req_min_uV) {
                int max_drop_uV = (uV * vctrl->ovp_threshold) / 100;
                next_uV = max_t(int, req_min_uV, uV - max_drop_uV);
                next_ctrl_uV = vctrl_calc_ctrl_voltage(vctrl, next_uV);
 
-               ret = regulator_set_voltage(ctrl_reg,
+               ret = regulator_set_voltage_rdev(ctrl_reg->rdev,
+                                           next_ctrl_uV,
                                            next_ctrl_uV,
-                                           next_ctrl_uV);
+                                           PM_SUSPEND_ON);
                if (ret)
                        goto err;
 
 
 err:
        /* Try to go back to original voltage */
-       regulator_set_voltage(ctrl_reg, orig_ctrl_uV, orig_ctrl_uV);
+       regulator_set_voltage_rdev(ctrl_reg->rdev, orig_ctrl_uV, orig_ctrl_uV,
+                                  PM_SUSPEND_ON);
 
        return ret;
 }
 
        if (selector >= vctrl->sel || !vctrl->ovp_threshold) {
                /* voltage rising or no OVP */
-               ret = regulator_set_voltage(ctrl_reg,
+               ret = regulator_set_voltage_rdev(ctrl_reg->rdev,
+                                           vctrl->vtable[selector].ctrl,
                                            vctrl->vtable[selector].ctrl,
-                                           vctrl->vtable[selector].ctrl);
+                                           PM_SUSPEND_ON);
                if (!ret)
                        vctrl->sel = selector;
 
                else
                        next_sel = vctrl->vtable[vctrl->sel].ovp_min_sel;
 
-               ret = regulator_set_voltage(ctrl_reg,
+               ret = regulator_set_voltage_rdev(ctrl_reg->rdev,
                                            vctrl->vtable[next_sel].ctrl,
-                                           vctrl->vtable[next_sel].ctrl);
+                                           vctrl->vtable[next_sel].ctrl,
+                                           PM_SUSPEND_ON);
                if (ret) {
                        dev_err(&rdev->dev,
                                "failed to set control voltage to %duV\n",
 err:
        if (vctrl->sel != orig_sel) {
                /* Try to go back to original voltage */
-               if (!regulator_set_voltage(ctrl_reg,
+               if (!regulator_set_voltage_rdev(ctrl_reg->rdev,
+                                          vctrl->vtable[orig_sel].ctrl,
                                           vctrl->vtable[orig_sel].ctrl,
-                                          vctrl->vtable[orig_sel].ctrl))
+                                          PM_SUSPEND_ON))
                        vctrl->sel = orig_sel;
                else
                        dev_warn(&rdev->dev,
                if (ret)
                        return ret;
 
-               ctrl_uV = regulator_get_voltage(vctrl->ctrl_reg);
+               ctrl_uV = regulator_get_voltage_rdev(vctrl->ctrl_reg->rdev);
                if (ctrl_uV < 0) {
                        dev_err(&pdev->dev, "failed to get control voltage\n");
                        return ctrl_uV;