mtd: {nand,spi-nor}: assign MTD of_node

We should pass along our flash DT node to the MTD layer, so it can set
up ofpart for us.

Signed-off-by: Brian Norris <computersforpeace@gmail.com>
Reviewed-by: Boris Brezillon <boris.brezillon@free-electrons.com>

diff --git a/drivers/mtd/nand/nand_base.c b/drivers/mtd/nand/nand_base.c
index cc74142938b0a21410cc76e0adca5a58aa1ce26f..939ab3d5acc2bec1ec7fa7b7f199227d78d8bae3 100644 (file)
--- a/drivers/mtd/nand/nand_base.c
+++ b/drivers/mtd/nand/nand_base.c
@@ -3990,6 +3990,9 @@ int nand_scan_ident(struct mtd_info *mtd, int maxchips,
        int ret;
 
        if (chip->flash_node) {
+               /* MTD can automatically handle DT partitions, etc. */
+               mtd_set_of_node(mtd, chip->flash_node);
+
                ret = nand_dt_init(mtd, chip, chip->flash_node);
                if (ret)
                        return ret;

diff --git a/drivers/mtd/spi-nor/spi-nor.c b/drivers/mtd/spi-nor/spi-nor.c
index 49883905a434b68b98af50e515a0356731c62fa6..1d3107ccee6164282abd46e78c36cd009a768231 100644 (file)
--- a/drivers/mtd/spi-nor/spi-nor.c
+++ b/drivers/mtd/spi-nor/spi-nor.c
@@ -1258,6 +1258,7 @@ int spi_nor_scan(struct spi_nor *nor, const char *name, enum read_mode mode)
                mtd->flags |= MTD_NO_ERASE;
 
        mtd->dev.parent = dev;
+       mtd_set_of_node(mtd, np);
        nor->page_size = info->page_size;
        mtd->writebufsize = nor->page_size;