1 files changed, 11 insertions, 12 deletions

diff --git a/drivers/power/avs/smartreflex.c b/drivers/power/avs/smartreflex.c
index 08a8a299f1fd..14cce7addbb9 100644
--- a/drivers/power/avs/smartreflex.c
+++ b/drivers/power/avs/smartreflex.c
@@ -552,7 +552,7 @@ int sr_configure_minmax(struct omap_sr *sr)
 
 /**
  * sr_enable() - Enables the smartreflex module.
- * @voltdm:     VDD pointer to which the SR module to be configured belongs to.
+ * @sr:         pointer to which the SR module to be configured belongs to.
  * @volt:       The voltage at which the Voltage domain associated with
  *              the smartreflex module is operating at.
  *              This is required only to program the correct Ntarget value.
@@ -561,16 +561,16 @@ int sr_configure_minmax(struct omap_sr *sr)
  * enable a smartreflex module. Returns 0 on success. Returns error
  * value if the voltage passed is wrong or if ntarget value is wrong.
  */
-int sr_enable(struct voltagedomain *voltdm, unsigned long volt)
+int sr_enable(struct omap_sr *sr, unsigned long volt)
 {
         struct omap_volt_data *volt_data;
-        struct omap_sr *sr = _sr_lookup(voltdm);
         struct omap_sr_nvalue_table *nvalue_row;
         int ret;
 
-        if (IS_ERR(sr)) {
-                pr_warning("%s: omap_sr struct for voltdm not found\n", __func__);
-                return PTR_ERR(sr);
+        if (!sr) {
+                pr_warn("%s: NULL omap_sr from %pF\n", __func__,
+                        (void *)_RET_IP_);
+                return -EINVAL;
         }
 
         volt_data = omap_voltage_get_voltdata(sr->voltdm, volt);
@@ -612,17 +612,16 @@ int sr_enable(struct voltagedomain *voltdm, unsigned long volt)
 
 /**
  * sr_disable() - Disables the smartreflex module.
- * @voltdm:     VDD pointer to which the SR module to be configured belongs to.
+ * @sr:         pointer to which the SR module to be configured belongs to.
  *
  * This API is to be called from the smartreflex class driver to
  * disable a smartreflex module.
  */
-void sr_disable(struct voltagedomain *voltdm)
+void sr_disable(struct omap_sr *sr)
 {
-        struct omap_sr *sr = _sr_lookup(voltdm);
-
-        if (IS_ERR(sr)) {
-                pr_warning("%s: omap_sr struct for voltdm not found\n", __func__);
+        if (!sr) {
+                pr_warn("%s: NULL omap_sr from %pF\n", __func__,
+                        (void *)_RET_IP_);
                 return;
         }