1 files changed, 7 insertions, 0 deletions

diff --git a/drivers/mmc/host/atmel-mci.c b/drivers/mmc/host/atmel-mci.c
index 945a31d77364..42706ea0ba85 100644
--- a/drivers/mmc/host/atmel-mci.c
+++ b/drivers/mmc/host/atmel-mci.c
@@ -1402,8 +1402,14 @@ static void atmci_set_ios(struct mmc_host *mmc, struct mmc_ios *ios)
                 clk_unprepare(host->mck);
 
         switch (ios->power_mode) {
+        case MMC_POWER_OFF:
+                if (!IS_ERR(mmc->supply.vmmc))
+                        mmc_regulator_set_ocr(mmc, mmc->supply.vmmc, 0);
+                break;
         case MMC_POWER_UP:
                 set_bit(ATMCI_CARD_NEED_INIT, &slot->flags);
+                if (!IS_ERR(mmc->supply.vmmc))
+                        mmc_regulator_set_ocr(mmc, mmc->supply.vmmc, ios->vdd);
                 break;
         default:
                 /*
@@ -2215,6 +2221,7 @@ static int __init atmci_init_slot(struct atmel_mci *host,
         }
 
         host->slot[id] = slot;
+        mmc_regulator_get_supply(mmc);
         mmc_add_host(mmc);
 
         if (gpio_is_valid(slot->detect_pin)) {