1 files changed, 30 insertions, 0 deletions

diff --git a/drivers/regulator/qcom_rpm-regulator.c b/drivers/regulator/qcom_rpm-regulator.c
index 00c5cc3d9546..15e07c233c1e 100644
--- a/drivers/regulator/qcom_rpm-regulator.c
+++ b/drivers/regulator/qcom_rpm-regulator.c
@@ -393,6 +393,28 @@ static int rpm_reg_is_enabled(struct regulator_dev *rdev)
         return vreg->is_enabled;
 }
 
+static int rpm_reg_set_load(struct regulator_dev *rdev, int load_uA)
+{
+        struct qcom_rpm_reg *vreg = rdev_get_drvdata(rdev);
+        const struct rpm_reg_parts *parts = vreg->parts;
+        const struct request_member *req = &parts->ia;
+        int load_mA = load_uA / 1000;
+        int max_mA = req->mask >> req->shift;
+        int ret;
+
+        if (req->mask == 0)
+                return -EINVAL;
+
+        if (load_mA > max_mA)
+                load_mA = max_mA;
+
+        mutex_lock(&vreg->lock);
+        ret = rpm_reg_write(vreg, req, load_mA);
+        mutex_unlock(&vreg->lock);
+
+        return ret;
+}
+
 static struct regulator_ops uV_ops = {
         .list_voltage = regulator_list_voltage_linear_range,
 
@@ -402,6 +424,8 @@ static struct regulator_ops uV_ops = {
         .enable = rpm_reg_uV_enable,
         .disable = rpm_reg_uV_disable,
         .is_enabled = rpm_reg_is_enabled,
+
+        .set_load = rpm_reg_set_load,
 };
 
 static struct regulator_ops mV_ops = {
@@ -413,6 +437,8 @@ static struct regulator_ops mV_ops = {
         .enable = rpm_reg_mV_enable,
         .disable = rpm_reg_mV_disable,
         .is_enabled = rpm_reg_is_enabled,
+
+        .set_load = rpm_reg_set_load,
 };
 
 static struct regulator_ops switch_ops = {
@@ -706,6 +732,10 @@ static int rpm_reg_probe(struct platform_device *pdev)
                 return -EINVAL;
         }
 
+        /* Regulators with ia property suppports drms */
+        if (vreg->parts->ia.mask)
+                initdata->constraints.valid_ops_mask |= REGULATOR_CHANGE_DRMS;
+
         key = "bias-pull-down";
         if (of_property_read_bool(pdev->dev.of_node, key)) {
                 ret = rpm_reg_set(vreg, &vreg->parts->pd, 1);