Manipulate clock scaling related stuff only if the host capability
supports clock scaling feature to avoid redundant code execution.

Signed-off-by: Stanley Chu <[email protected]>
---
 drivers/scsi/ufs/ufshcd.c | 54 ++++++++++++++++++++++-----------------
 1 file changed, 30 insertions(+), 24 deletions(-)

diff --git a/drivers/scsi/ufs/ufshcd.c b/drivers/scsi/ufs/ufshcd.c
index 9cc16598136d..15e5877411b3 100644
--- a/drivers/scsi/ufs/ufshcd.c
+++ b/drivers/scsi/ufs/ufshcd.c
@@ -1448,9 +1448,6 @@ static void ufshcd_suspend_clkscaling(struct ufs_hba *hba)
        unsigned long flags;
        bool suspend = false;
 
-       if (!ufshcd_is_clkscaling_supported(hba))
-               return;
-
        cancel_work_sync(&hba->clk_scaling.suspend_work);
        cancel_work_sync(&hba->clk_scaling.resume_work);
 
@@ -1470,9 +1467,6 @@ static void ufshcd_resume_clkscaling(struct ufs_hba *hba)
        unsigned long flags;
        bool resume = false;
 
-       if (!ufshcd_is_clkscaling_supported(hba))
-               return;
-
        spin_lock_irqsave(hba->host->host_lock, flags);
        if (hba->clk_scaling.is_suspended) {
                resume = true;
@@ -5663,7 +5657,8 @@ static void ufshcd_err_handling_prepare(struct ufs_hba 
*hba)
                ufshcd_vops_resume(hba, UFS_RUNTIME_PM);
        } else {
                ufshcd_hold(hba, false);
-               if (hba->clk_scaling.is_enabled)
+               if (ufshcd_is_clkscaling_supported(hba) &&
+                   hba->clk_scaling.is_enabled)
                        ufshcd_suspend_clkscaling(hba);
        }
        down_write(&hba->clk_scaling_lock);
@@ -5678,7 +5673,8 @@ static void ufshcd_err_handling_unprepare(struct ufs_hba 
*hba)
        down_write(&hba->clk_scaling_lock);
        hba->clk_scaling.is_allowed = true;
        up_write(&hba->clk_scaling_lock);
-       if (hba->clk_scaling.is_enabled)
+       if (ufshcd_is_clkscaling_supported(hba) &&
+           hba->clk_scaling.is_enabled)
                ufshcd_resume_clkscaling(hba);
        pm_runtime_put(hba->dev);
 }
@@ -8466,6 +8462,25 @@ static void ufshcd_hba_vreg_set_hpm(struct ufs_hba *hba)
                ufshcd_setup_hba_vreg(hba, true);
 }
 
+static void ufshcd_clk_scaling_pm(struct ufs_hba *hba, bool suspend)
+{
+       if (suspend) {
+               if (hba->clk_scaling.is_enabled)
+                       ufshcd_suspend_clkscaling(hba);
+
+               down_write(&hba->clk_scaling_lock);
+               hba->clk_scaling.is_allowed = false;
+               up_write(&hba->clk_scaling_lock);
+       } else {
+               down_write(&hba->clk_scaling_lock);
+               hba->clk_scaling.is_allowed = true;
+               up_write(&hba->clk_scaling_lock);
+
+               if (hba->clk_scaling.is_enabled)
+                       ufshcd_resume_clkscaling(hba);
+       }
+}
+
 /**
  * ufshcd_suspend - helper function for suspend operations
  * @hba: per adapter instance
@@ -8507,12 +8522,8 @@ static int ufshcd_suspend(struct ufs_hba *hba, enum 
ufs_pm_op pm_op)
        ufshcd_hold(hba, false);
        hba->clk_gating.is_suspended = true;
 
-       if (hba->clk_scaling.is_enabled)
-               ufshcd_suspend_clkscaling(hba);
-
-       down_write(&hba->clk_scaling_lock);
-       hba->clk_scaling.is_allowed = false;
-       up_write(&hba->clk_scaling_lock);
+       if (ufshcd_is_clkscaling_supported(hba))
+               ufshcd_clk_scaling_pm(hba, true);
 
        if (req_dev_pwr_mode == UFS_ACTIVE_PWR_MODE &&
                        req_link_state == UIC_LINK_ACTIVE_STATE) {
@@ -8618,11 +8629,9 @@ static int ufshcd_suspend(struct ufs_hba *hba, enum 
ufs_pm_op pm_op)
        if (!ufshcd_set_dev_pwr_mode(hba, UFS_ACTIVE_PWR_MODE))
                ufshcd_disable_auto_bkops(hba);
 enable_gating:
-       down_write(&hba->clk_scaling_lock);
-       hba->clk_scaling.is_allowed = true;
-       up_write(&hba->clk_scaling_lock);
-       if (hba->clk_scaling.is_enabled)
-               ufshcd_resume_clkscaling(hba);
+       if (ufshcd_is_clkscaling_supported(hba))
+               ufshcd_clk_scaling_pm(hba, false);
+
        hba->clk_gating.is_suspended = false;
        hba->dev_info.b_rpm_dev_flush_capable = false;
        ufshcd_release(hba);
@@ -8719,11 +8728,8 @@ static int ufshcd_resume(struct ufs_hba *hba, enum 
ufs_pm_op pm_op)
 
        hba->clk_gating.is_suspended = false;
 
-       down_write(&hba->clk_scaling_lock);
-       hba->clk_scaling.is_allowed = true;
-       up_write(&hba->clk_scaling_lock);
-       if (hba->clk_scaling.is_enabled)
-               ufshcd_resume_clkscaling(hba);
+       if (ufshcd_is_clkscaling_supported(hba))
+               ufshcd_clk_scaling_pm(hba, false);
 
        /* Enable Auto-Hibernate if configured */
        ufshcd_auto_hibern8_enable(hba);
-- 
2.18.0

Reply via email to