int min_uV, int max_uV);
 static int regulator_balance_voltage(struct regulator_dev *rdev,
                                     suspend_state_t state);
+static int regulator_set_voltage_rdev(struct regulator_dev *rdev,
+                                     int min_uV, int max_uV,
+                                     suspend_state_t state);
 static struct regulator *create_regulator(struct regulator_dev *rdev,
                                          struct device *dev,
                                          const char *supply_name);
        }
 }
 
-/**
- * regulator_lock_supply - lock a regulator and its supplies
- * @rdev:         regulator source
- */
-static void regulator_lock_supply(struct regulator_dev *rdev)
+static int regulator_lock_recursive(struct regulator_dev *rdev,
+                                   unsigned int subclass)
 {
+       struct regulator_dev *c_rdev;
        int i;
 
-       for (i = 0; rdev; rdev = rdev_get_supply(rdev), i++)
-               regulator_lock_nested(rdev, i);
+       for (i = 0; i < rdev->coupling_desc.n_coupled; i++) {
+               c_rdev = rdev->coupling_desc.coupled_rdevs[i];
+
+               if (!c_rdev)
+                       continue;
+
+               regulator_lock_nested(c_rdev, subclass++);
+
+               if (c_rdev->supply)
+                       subclass =
+                               regulator_lock_recursive(c_rdev->supply->rdev,
+                                                        subclass);
+       }
+
+       return subclass;
 }
 
 /**
- * regulator_unlock_supply - unlock a regulator and its supplies
- * @rdev:         regulator source
+ * regulator_unlock_dependent - unlock regulator's suppliers and coupled
+ *                             regulators
+ * @rdev:                      regulator source
+ *
+ * Unlock all regulators related with rdev by coupling or suppling.
  */
-static void regulator_unlock_supply(struct regulator_dev *rdev)
+static void regulator_unlock_dependent(struct regulator_dev *rdev)
 {
-       struct regulator *supply;
+       struct regulator_dev *c_rdev;
+       int i;
 
-       while (1) {
-               regulator_unlock(rdev);
-               supply = rdev->supply;
+       for (i = 0; i < rdev->coupling_desc.n_coupled; i++) {
+               c_rdev = rdev->coupling_desc.coupled_rdevs[i];
 
-               if (!rdev->supply)
-                       return;
+               if (!c_rdev)
+                       continue;
 
-               rdev = supply->rdev;
+               regulator_unlock(c_rdev);
+
+               if (c_rdev->supply)
+                       regulator_unlock_dependent(c_rdev->supply->rdev);
        }
 }
 
+/**
+ * regulator_lock_dependent - lock regulator's suppliers and coupled regulators
+ * @rdev:                      regulator source
+ *
+ * This function as a wrapper on regulator_lock_recursive(), which locks
+ * all regulators related with rdev by coupling or suppling.
+ */
+static inline void regulator_lock_dependent(struct regulator_dev *rdev)
+{
+       regulator_lock_recursive(rdev, 0);
+}
+
 /**
  * of_get_regulator - get a regulator device node based on supply name
  * @dev: Device pointer for the consumer (of regulator) device
                        return ret;
        }
 
-       mutex_lock(&rdev->mutex);
+       regulator_lock_dependent(rdev);
+       /* balance only if there are regulators coupled */
+       if (rdev->coupling_desc.n_coupled > 1) {
+               ret = regulator_balance_voltage(rdev, PM_SUSPEND_ON);
+               if (ret != 0)
+                       goto unlock;
+       }
        ret = _regulator_enable(rdev);
-       mutex_unlock(&rdev->mutex);
+unlock:
+       regulator_unlock_dependent(rdev);
 
        if (ret != 0 && rdev->supply)
                regulator_disable(rdev->supply);
        if (regulator->always_on)
                return 0;
 
-       mutex_lock(&rdev->mutex);
+       regulator_lock_dependent(rdev);
        ret = _regulator_disable(rdev);
-       mutex_unlock(&rdev->mutex);
+       if (rdev->coupling_desc.n_coupled > 1)
+               regulator_balance_voltage(rdev, PM_SUSPEND_ON);
+       regulator_unlock_dependent(rdev);
 
        if (ret == 0 && rdev->supply)
                regulator_disable(rdev->supply);
        struct regulator_dev *rdev = regulator->rdev;
        int ret;
 
-       mutex_lock(&rdev->mutex);
+       regulator_lock_dependent(rdev);
        regulator->uA_load = 0;
        ret = _regulator_force_disable(regulator->rdev);
-       mutex_unlock(&rdev->mutex);
+       if (rdev->coupling_desc.n_coupled > 1)
+               regulator_balance_voltage(rdev, PM_SUSPEND_ON);
+       regulator_unlock_dependent(rdev);
 
        if (rdev->supply)
                while (rdev->open_count--)
        if (regulator->always_on)
                return 1;
 
-       mutex_lock(®ulator->rdev->mutex);
+       regulator_lock_dependent(regulator->rdev);
        ret = _regulator_is_enabled(regulator->rdev);
-       mutex_unlock(®ulator->rdev->mutex);
+       regulator_unlock_dependent(regulator->rdev);
 
        return ret;
 }
        int ret = 0;
        int old_min_uV, old_max_uV;
        int current_uV;
-       int best_supply_uV = 0;
-       int supply_change_uV = 0;
 
        /* If we're setting the same range as last time the change
         * should be a noop (some cpufreq implementations use the same
        voltage->min_uV = min_uV;
        voltage->max_uV = max_uV;
 
-       ret = regulator_check_consumers(rdev, &min_uV, &max_uV, state);
+       /* for not coupled regulators this will just set the voltage */
+       ret = regulator_balance_voltage(rdev, state);
        if (ret < 0)
                goto out2;
 
+out:
+       return 0;
+out2:
+       voltage->min_uV = old_min_uV;
+       voltage->max_uV = old_max_uV;
+
+       return ret;
+}
+
+static int regulator_set_voltage_rdev(struct regulator_dev *rdev, int min_uV,
+                                     int max_uV, suspend_state_t state)
+{
+       int best_supply_uV = 0;
+       int supply_change_uV = 0;
+       int ret;
+
        if (rdev->supply &&
            regulator_ops_is_valid(rdev->supply->rdev,
                                   REGULATOR_CHANGE_VOLTAGE) &&
                selector = regulator_map_voltage(rdev, min_uV, max_uV);
                if (selector < 0) {
                        ret = selector;
-                       goto out2;
+                       goto out;
                }
 
                best_supply_uV = _regulator_list_voltage(rdev, selector, 0);
                if (best_supply_uV < 0) {
                        ret = best_supply_uV;
-                       goto out2;
+                       goto out;
                }
 
                best_supply_uV += rdev->desc->min_dropout_uV;
                current_supply_uV = _regulator_get_voltage(rdev->supply->rdev);
                if (current_supply_uV < 0) {
                        ret = current_supply_uV;
-                       goto out2;
+                       goto out;
                }
 
                supply_change_uV = best_supply_uV - current_supply_uV;
                if (ret) {
                        dev_err(&rdev->dev, "Failed to increase supply voltage: %d\n",
                                        ret);
-                       goto out2;
+                       goto out;
                }
        }
 
                ret = _regulator_do_set_suspend_voltage(rdev, min_uV,
                                                        max_uV, state);
        if (ret < 0)
-               goto out2;
+               goto out;
 
        if (supply_change_uV < 0) {
                ret = regulator_set_voltage_unlocked(rdev->supply,
        }
 
 out:
-       return ret;
-out2:
-       voltage->min_uV = old_min_uV;
-       voltage->max_uV = old_max_uV;
-
        return ret;
 }
 
                        ret = 0;
                        goto out;
                }
-#if 0
+
                ret = regulator_set_voltage_rdev(best_rdev, best_min_uV,
                                                 best_max_uV, state);
-#endif
+
                if (ret < 0)
                        goto out;
 
 {
        int ret = 0;
 
-       regulator_lock_supply(regulator->rdev);
+       regulator_lock_dependent(regulator->rdev);
 
        ret = regulator_set_voltage_unlocked(regulator, min_uV, max_uV,
                                             PM_SUSPEND_ON);
 
-       regulator_unlock_supply(regulator->rdev);
+       regulator_unlock_dependent(regulator->rdev);
 
        return ret;
 }
        if (regulator_check_states(state) || state == PM_SUSPEND_ON)
                return -EINVAL;
 
-       regulator_lock_supply(regulator->rdev);
+       regulator_lock_dependent(regulator->rdev);
 
        ret = _regulator_set_suspend_voltage(regulator, min_uV,
                                             max_uV, state);
 
-       regulator_unlock_supply(regulator->rdev);
+       regulator_unlock_dependent(regulator->rdev);
 
        return ret;
 }
 {
        int ret;
 
-       regulator_lock_supply(regulator->rdev);
+       regulator_lock_dependent(regulator->rdev);
 
        ret = _regulator_get_voltage(regulator->rdev);
 
-       regulator_unlock_supply(regulator->rdev);
+       regulator_unlock_dependent(regulator->rdev);
 
        return ret;
 }