1 --- a/drivers/mtd/mtk-snand/mtk-snand-mtd.c
2 +++ b/drivers/mtd/mtk-snand/mtk-snand-mtd.c
4 #include <linux/dma-mapping.h>
5 #include <linux/wait.h>
6 #include <linux/mtd/mtd.h>
7 +#include <linux/mtd/mtk_bmt.h>
8 #include <linux/mtd/partitions.h>
9 #include <linux/of_platform.h>
11 @@ -612,6 +613,8 @@ static int mtk_snand_probe(struct platfo
12 mtd->_block_isbad = mtk_snand_mtd_block_isbad;
13 mtd->_block_markbad = mtk_snand_mtd_block_markbad;
15 + mtk_bmt_attach(mtd);
17 ret = mtd_device_register(mtd, NULL, 0);
19 dev_err(msm->pdev.dev, "failed to register mtd partition\n");
20 @@ -623,6 +626,7 @@ static int mtk_snand_probe(struct platfo
24 + mtk_bmt_detach(mtd);
25 devm_kfree(msm->pdev.dev, msm->page_cache);
28 @@ -650,6 +654,8 @@ static int mtk_snand_remove(struct platf
32 + mtk_bmt_detach(mtd);
34 mtk_snand_cleanup(msm->snf);