return 0;
 }
 
-static void ufs_mtk_vreg_set_lpm(struct ufs_hba *hba, bool lpm)
+static void ufs_mtk_vccqx_set_lpm(struct ufs_hba *hba, bool lpm)
 {
        struct ufs_vreg *vccqx = NULL;
 
+       if (hba->vreg_info.vccq)
+               vccqx = hba->vreg_info.vccq;
+       else
+               vccqx = hba->vreg_info.vccq2;
+
+       regulator_set_mode(vccqx->reg,
+                          lpm ? REGULATOR_MODE_IDLE : REGULATOR_MODE_NORMAL);
+}
+
+static void ufs_mtk_vsx_set_lpm(struct ufs_hba *hba, bool lpm)
+{
+       struct arm_smccc_res res;
+
+       ufs_mtk_device_pwr_ctrl(!lpm,
+                               (unsigned long)hba->dev_info.wspecversion,
+                               res);
+}
+
+static void ufs_mtk_dev_vreg_set_lpm(struct ufs_hba *hba, bool lpm)
+{
        if (!hba->vreg_info.vccq && !hba->vreg_info.vccq2)
                return;
 
        if (lpm && hba->vreg_info.vcc->enabled)
                return;
 
-       if (hba->vreg_info.vccq)
-               vccqx = hba->vreg_info.vccq;
-       else
-               vccqx = hba->vreg_info.vccq2;
-
-       regulator_set_mode(vccqx->reg,
-                          lpm ? REGULATOR_MODE_IDLE : REGULATOR_MODE_NORMAL);
+       if (lpm) {
+               ufs_mtk_vccqx_set_lpm(hba, lpm);
+               ufs_mtk_vsx_set_lpm(hba, lpm);
+       } else {
+               ufs_mtk_vsx_set_lpm(hba, lpm);
+               ufs_mtk_vccqx_set_lpm(hba, lpm);
+       }
 }
 
 static void ufs_mtk_auto_hibern8_disable(struct ufs_hba *hba)
        int err;
 
        if (hba->ufshcd_state != UFSHCD_STATE_OPERATIONAL)
-               ufs_mtk_vreg_set_lpm(hba, false);
+               ufs_mtk_dev_vreg_set_lpm(hba, false);
 
        err = ufs_mtk_mphy_power_on(hba, true);
        if (err)
        if (ret)
                return ret;
 
-       ufs_mtk_vreg_set_lpm(hba, true);
+       ufs_mtk_dev_vreg_set_lpm(hba, true);
 
        return 0;
 }
 {
        struct ufs_hba *hba = dev_get_drvdata(dev);
 
-       ufs_mtk_vreg_set_lpm(hba, false);
+       ufs_mtk_dev_vreg_set_lpm(hba, false);
 
        return ufshcd_system_resume(dev);
 }
        if (ret)
                return ret;
 
-       ufs_mtk_vreg_set_lpm(hba, true);
+       ufs_mtk_dev_vreg_set_lpm(hba, true);
 
        return 0;
 }
 {
        struct ufs_hba *hba = dev_get_drvdata(dev);
 
-       ufs_mtk_vreg_set_lpm(hba, false);
+       ufs_mtk_dev_vreg_set_lpm(hba, false);
 
        return ufshcd_runtime_resume(dev);
 }
 
 #define UFS_MTK_SIP_DEVICE_RESET          BIT(1)
 #define UFS_MTK_SIP_CRYPTO_CTRL           BIT(2)
 #define UFS_MTK_SIP_REF_CLK_NOTIFICATION  BIT(3)
+#define UFS_MTK_SIP_DEVICE_PWR_CTRL       BIT(7)
 
 /*
  * VS_DEBUGCLOCKENABLE
 #define ufs_mtk_device_reset_ctrl(high, res) \
        ufs_mtk_smc(UFS_MTK_SIP_DEVICE_RESET, &(res), high)
 
+#define ufs_mtk_device_pwr_ctrl(on, ufs_ver, res) \
+       ufs_mtk_smc(UFS_MTK_SIP_DEVICE_PWR_CTRL, &(res), on, ufs_ver)
+
 #endif /* !_UFS_MEDIATEK_H */