1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/mmc/core/mmc.c b/drivers/mmc/core/mmc.c
index da892a599524..09c688f5ff65 100644
--- a/drivers/mmc/core/mmc.c
+++ b/drivers/mmc/core/mmc.c
@@ -1743,6 +1743,14 @@ static int mmc_init_card(struct mmc_host *host, u32 ocr,
                         card->ext_csd.power_off_notification = EXT_CSD_POWER_ON;
         }
 
+        /* set erase_arg */
+        if (mmc_can_discard(card))
+                card->erase_arg = MMC_DISCARD_ARG;
+        else if (mmc_can_trim(card))
+                card->erase_arg = MMC_TRIM_ARG;
+        else
+                card->erase_arg = MMC_ERASE_ARG;
+
         /*
          * Select timing interface
          */