}
 
 /**
- * dev_pm_opp_register_set_opp_helper() - Register custom set OPP helper
+ * _opp_register_set_opp_helper() - Register custom set OPP helper
  * @dev: Device for which the helper is getting registered.
  * @set_opp: Custom set OPP helper.
  *
  *
  * This must be called before any OPPs are initialized for the device.
  */
-struct opp_table *dev_pm_opp_register_set_opp_helper(struct device *dev,
-                       int (*set_opp)(struct dev_pm_set_opp_data *data))
+static int _opp_register_set_opp_helper(struct opp_table *opp_table,
+       struct device *dev, int (*set_opp)(struct dev_pm_set_opp_data *data))
 {
        struct dev_pm_set_opp_data *data;
-       struct opp_table *opp_table;
-
-       if (!set_opp)
-               return ERR_PTR(-EINVAL);
-
-       opp_table = _add_opp_table(dev, false);
-       if (IS_ERR(opp_table))
-               return opp_table;
-
-       /* This should be called before OPPs are initialized */
-       if (WARN_ON(!list_empty(&opp_table->opp_list))) {
-               dev_pm_opp_put_opp_table(opp_table);
-               return ERR_PTR(-EBUSY);
-       }
 
        /* Another CPU that shares the OPP table has set the helper ? */
        if (opp_table->set_opp)
-               return opp_table;
+               return 0;
 
        data = kzalloc(sizeof(*data), GFP_KERNEL);
        if (!data)
-               return ERR_PTR(-ENOMEM);
+               return -ENOMEM;
 
        mutex_lock(&opp_table->lock);
        opp_table->set_opp_data = data;
 
        opp_table->set_opp = set_opp;
 
-       return opp_table;
+       return 0;
 }
-EXPORT_SYMBOL_GPL(dev_pm_opp_register_set_opp_helper);
 
 /**
- * dev_pm_opp_unregister_set_opp_helper() - Releases resources blocked for
- *                                        set_opp helper
- * @opp_table: OPP table returned from dev_pm_opp_register_set_opp_helper().
+ * _opp_unregister_set_opp_helper() - Releases resources blocked for set_opp helper
+ * @opp_table: OPP table returned from _opp_register_set_opp_helper().
  *
  * Release resources blocked for platform specific set_opp helper.
  */
-void dev_pm_opp_unregister_set_opp_helper(struct opp_table *opp_table)
+static void _opp_unregister_set_opp_helper(struct opp_table *opp_table)
 {
-       if (unlikely(!opp_table))
-               return;
-
-       opp_table->set_opp = NULL;
-
-       mutex_lock(&opp_table->lock);
-       kfree(opp_table->set_opp_data);
-       opp_table->set_opp_data = NULL;
-       mutex_unlock(&opp_table->lock);
-
-       dev_pm_opp_put_opp_table(opp_table);
-}
-EXPORT_SYMBOL_GPL(dev_pm_opp_unregister_set_opp_helper);
-
-static void devm_pm_opp_unregister_set_opp_helper(void *data)
-{
-       dev_pm_opp_unregister_set_opp_helper(data);
-}
-
-/**
- * devm_pm_opp_register_set_opp_helper() - Register custom set OPP helper
- * @dev: Device for which the helper is getting registered.
- * @set_opp: Custom set OPP helper.
- *
- * This is a resource-managed version of dev_pm_opp_register_set_opp_helper().
- *
- * Return: 0 on success and errorno otherwise.
- */
-int devm_pm_opp_register_set_opp_helper(struct device *dev,
-                                       int (*set_opp)(struct dev_pm_set_opp_data *data))
-{
-       struct opp_table *opp_table;
-
-       opp_table = dev_pm_opp_register_set_opp_helper(dev, set_opp);
-       if (IS_ERR(opp_table))
-               return PTR_ERR(opp_table);
+       if (opp_table->set_opp) {
+               opp_table->set_opp = NULL;
 
-       return devm_add_action_or_reset(dev, devm_pm_opp_unregister_set_opp_helper,
-                                       opp_table);
+               mutex_lock(&opp_table->lock);
+               kfree(opp_table->set_opp_data);
+               opp_table->set_opp_data = NULL;
+               mutex_unlock(&opp_table->lock);
+       }
 }
-EXPORT_SYMBOL_GPL(devm_pm_opp_register_set_opp_helper);
 
 static void _opp_detach_genpd(struct opp_table *opp_table)
 {
        if (data->flags & OPP_CONFIG_SUPPORTED_HW)
                _opp_put_supported_hw(data->opp_table);
        if (data->flags & OPP_CONFIG_REGULATOR_HELPER)
-               dev_pm_opp_unregister_set_opp_helper(data->opp_table);
+               _opp_unregister_set_opp_helper(data->opp_table);
        if (data->flags & OPP_CONFIG_PROP_NAME)
                dev_pm_opp_put_prop_name(data->opp_table);
        if (data->flags & OPP_CONFIG_CLK)
 
        /* Configure opp helper */
        if (config->set_opp) {
-               err = dev_pm_opp_register_set_opp_helper(dev, config->set_opp);
-               if (IS_ERR(err)) {
-                       ret = PTR_ERR(err);
+               ret = _opp_register_set_opp_helper(opp_table, dev,
+                                                  config->set_opp);
+               if (ret)
                        goto err;
-               }
 
                data->flags |= OPP_CONFIG_REGULATOR_HELPER;
        }
 
 
 struct opp_table *dev_pm_opp_set_prop_name(struct device *dev, const char *name);
 void dev_pm_opp_put_prop_name(struct opp_table *opp_table);
-struct opp_table *dev_pm_opp_register_set_opp_helper(struct device *dev, int (*set_opp)(struct dev_pm_set_opp_data *data));
-void dev_pm_opp_unregister_set_opp_helper(struct opp_table *opp_table);
-int devm_pm_opp_register_set_opp_helper(struct device *dev, int (*set_opp)(struct dev_pm_set_opp_data *data));
 struct opp_table *dev_pm_opp_attach_genpd(struct device *dev, const char * const *names, struct device ***virt_devs);
 void dev_pm_opp_detach_genpd(struct opp_table *opp_table);
 int devm_pm_opp_attach_genpd(struct device *dev, const char * const *names, struct device ***virt_devs);
        return -EOPNOTSUPP;
 }
 
-static inline struct opp_table *dev_pm_opp_register_set_opp_helper(struct device *dev,
-                       int (*set_opp)(struct dev_pm_set_opp_data *data))
-{
-       return ERR_PTR(-EOPNOTSUPP);
-}
-
-static inline void dev_pm_opp_unregister_set_opp_helper(struct opp_table *opp_table) {}
-
-static inline int devm_pm_opp_register_set_opp_helper(struct device *dev,
-                                   int (*set_opp)(struct dev_pm_set_opp_data *data))
-{
-       return -EOPNOTSUPP;
-}
-
 static inline struct opp_table *dev_pm_opp_set_prop_name(struct device *dev, const char *name)
 {
        return ERR_PTR(-EOPNOTSUPP);
        return devm_pm_opp_set_config(dev, &config);
 }
 
+/* set-opp helpers */
+static inline int dev_pm_opp_register_set_opp_helper(struct device *dev,
+                       int (*set_opp)(struct dev_pm_set_opp_data *data))
+{
+       struct dev_pm_opp_config config = {
+               .set_opp = set_opp,
+       };
+
+       return dev_pm_opp_set_config(dev, &config);
+}
+
+static inline void dev_pm_opp_unregister_set_opp_helper(int token)
+{
+       dev_pm_opp_clear_config(token);
+}
+
 #endif         /* __LINUX_OPP_H__ */