ata: ahci: Enable/disable partial or slumber

Add options to enable/disable partial or slumber

Bug 200123901

Signed-off-by: Preetham Chandru R <pchandru@nvidia.com>
Signed-off-by: Seema Khowala <seemaj@nvidia.com>
Change-Id: I6ae6d8c3d58796931e9cb5252488cb69575b0440
Reviewed-on: http://git-master/r/1127297
(cherry picked from commit 8429e54ed40570fbc7797f33e1ea4e3b528cdd27)

1 files changed, 22 insertions, 0 deletions

diff --git a/drivers/ata/tegra/ahci_tegra.c b/drivers/ata/tegra/ahci_tegra.c
index c22baf78a..03a47540c 100644
--- a/drivers/ata/tegra/ahci_tegra.c
+++ b/drivers/ata/tegra/ahci_tegra.c
@@ -768,6 +768,24 @@ static void tegra_ahci_disable_hipm(struct ahci_host_priv *hpriv)
         tegra_ahci_scfg_update(hpriv, val, mask, T_SATA0_AHCI_HBA_CAP_BKDR);
 }
 
+static void tegra_ahci_disable_partial(struct ahci_host_priv *hpriv)
+{
+        u32 val = 0;
+        u32 mask = T_SATA0_AHCI_HBA_CAP_BKDR_PARTIAL_ST_CAP;
+
+        val = 0xFFFFFFFF & ~T_SATA0_AHCI_HBA_CAP_BKDR_PARTIAL_ST_CAP;
+        tegra_ahci_scfg_update(hpriv, val, mask, T_SATA0_AHCI_HBA_CAP_BKDR);
+}
+
+static void tegra_ahci_disable_slumber(struct ahci_host_priv *hpriv)
+{
+        u32 val = 0;
+        u32 mask = T_SATA0_AHCI_HBA_CAP_BKDR_SLUMBER_ST_CAP;
+
+        val = 0xFFFFFFFF & ~T_SATA0_AHCI_HBA_CAP_BKDR_SLUMBER_ST_CAP;
+        tegra_ahci_scfg_update(hpriv, val, mask, T_SATA0_AHCI_HBA_CAP_BKDR);
+}
+
 static void tegra_ahci_disable_ncq(struct ahci_host_priv *hpriv)
 {
         u32 val = 0;
@@ -802,6 +820,10 @@ static int tegra_ahci_disable_features(struct ahci_host_priv *hpriv)
                         tegra_ahci_disable_ncq(hpriv);
                 } else if (!strcmp(feature, "dipm")) {
                         ahci_tegra_port_info.flags |= ATA_FLAG_NO_DIPM;
+                } else if (!strcmp(feature, "partial")) {
+                        tegra_ahci_disable_partial(hpriv);
+                } else if (!strcmp(feature, "slumber")) {
+                        tegra_ahci_disable_slumber(hpriv);
                 }
         }