@@ -177,14 +177,26 @@ static int txgbe_i2c_request_regs(struct dw_i2c_dev *dev)
return 0;
}
-static void dw_i2c_plat_pm_cleanup(struct dw_i2c_dev *dev)
+static void dw_i2c_plat_pm_cleanup(void *data)
{
+ struct dw_i2c_dev *dev = data;
+
pm_runtime_disable(dev->dev);
if (dev->shared_with_punit)
pm_runtime_put_noidle(dev->dev);
}
+static int dw_i2c_plat_pm_setup(struct dw_i2c_dev *dev)
+{
+ if (dev->shared_with_punit)
+ pm_runtime_get_noresume(dev->dev);
+
+ pm_runtime_enable(dev->dev);
+
+ return devm_add_action_or_reset(dev->dev, dw_i2c_plat_pm_cleanup, dev);
+}
+
static int dw_i2c_plat_request_regs(struct dw_i2c_dev *dev)
{
struct platform_device *pdev = to_platform_device(dev->dev);
@@ -381,19 +393,16 @@ static int dw_i2c_plat_probe(struct platform_device *pdev)
pm_runtime_use_autosuspend(&pdev->dev);
pm_runtime_set_active(&pdev->dev);
- if (dev->shared_with_punit)
- pm_runtime_get_noresume(&pdev->dev);
-
- pm_runtime_enable(&pdev->dev);
+ ret = dw_i2c_plat_pm_setup(dev);
+ if (ret)
+ goto exit_reset;
ret = i2c_dw_probe(dev);
if (ret)
- goto exit_probe;
+ goto exit_reset;
return ret;
-exit_probe:
- dw_i2c_plat_pm_cleanup(dev);
exit_reset:
reset_control_assert(dev->rst);
return ret;
@@ -411,7 +420,6 @@ static void dw_i2c_plat_remove(struct platform_device *pdev)
pm_runtime_dont_use_autosuspend(&pdev->dev);
pm_runtime_put_sync(&pdev->dev);
- dw_i2c_plat_pm_cleanup(dev);
i2c_dw_remove_lock_support(dev);