1 files changed, 6 insertions, 16 deletions

diff --git a/drivers/base/power/opp/core.c b/drivers/base/power/opp/core.c
index 4c7c6da7a989..2824d3a5e9f0 100644
--- a/drivers/base/power/opp/core.c
+++ b/drivers/base/power/opp/core.c
@@ -1316,7 +1316,7 @@ EXPORT_SYMBOL_GPL(dev_pm_opp_put_prop_name);
  * that this function is *NOT* called under RCU protection or in contexts where
  * mutex cannot be locked.
  */
-int dev_pm_opp_set_regulator(struct device *dev, const char *name)
+struct opp_table *dev_pm_opp_set_regulator(struct device *dev, const char *name)
 {
         struct opp_table *opp_table;
         struct regulator *reg;
@@ -1354,20 +1354,20 @@ int dev_pm_opp_set_regulator(struct device *dev, const char *name)
         opp_table->regulator = reg;
 
         mutex_unlock(&opp_table_lock);
-        return 0;
+        return opp_table;
 
 err:
         _remove_opp_table(opp_table);
 unlock:
         mutex_unlock(&opp_table_lock);
 
-        return ret;
+        return ERR_PTR(ret);
 }
 EXPORT_SYMBOL_GPL(dev_pm_opp_set_regulator);
 
 /**
  * dev_pm_opp_put_regulator() - Releases resources blocked for regulator
- * @dev: Device for which regulator was set.
+ * @opp_table: OPP table returned from dev_pm_opp_set_regulator().
  *
  * Locking: The internal opp_table and opp structures are RCU protected.
  * Hence this function internally uses RCU updater strategy with mutex locks
@@ -1375,22 +1375,12 @@ EXPORT_SYMBOL_GPL(dev_pm_opp_set_regulator);
  * that this function is *NOT* called under RCU protection or in contexts where
  * mutex cannot be locked.
  */
-void dev_pm_opp_put_regulator(struct device *dev)
+void dev_pm_opp_put_regulator(struct opp_table *opp_table)
 {
-        struct opp_table *opp_table;
-
         mutex_lock(&opp_table_lock);
 
-        /* Check for existing table for 'dev' first */
-        opp_table = _find_opp_table(dev);
-        if (IS_ERR(opp_table)) {
-                dev_err(dev, "Failed to find opp_table: %ld\n",
-                        PTR_ERR(opp_table));
-                goto unlock;
-        }
-
         if (IS_ERR(opp_table->regulator)) {
-                dev_err(dev, "%s: Doesn't have regulator set\n", __func__);
+                pr_err("%s: Doesn't have regulator set\n", __func__);
                 goto unlock;
         }