1 files changed, 4 insertions, 0 deletions

diff --git a/drivers/mmc/core/bus.c b/drivers/mmc/core/bus.c
index 86d271148528..c5ef10065a4a 100644
--- a/drivers/mmc/core/bus.c
+++ b/drivers/mmc/core/bus.c
@@ -16,6 +16,7 @@
 #include <linux/err.h>
 #include <linux/slab.h>
 #include <linux/stat.h>
+#include <linux/of.h>
 #include <linux/pm_runtime.h>
 
 #include <linux/mmc/card.h>
@@ -321,6 +322,8 @@ int mmc_add_card(struct mmc_card *card)
 #endif
         mmc_init_context_info(card->host);
 
+        card->dev.of_node = mmc_of_find_child_device(card->host, 0);
+
         ret = device_add(&card->dev);
         if (ret)
                 return ret;
@@ -349,6 +352,7 @@ void mmc_remove_card(struct mmc_card *card)
                                 mmc_hostname(card->host), card->rca);
                 }
                 device_del(&card->dev);
+                of_node_put(card->dev.of_node);
         }
 
         put_device(&card->dev);