commit 06419a4bae2adcdd0ed4e73f93f3f40a82172e45
author Googler <noreply@google.com> Tue May 17 21:32:34 2022 +0800
committer Googler <noreply@google.com> Thu Jan 05 15:57:53 2023 -0800
tree bfbdd1a698d8e19d52612ee1aec20b616248dd72
parent e00b8eb55a8ecd6051b13309a1b5aa88400ba74d

Project import generated by Copybara.

GitOrigin-RevId: ceda9993b5e33a3605cf30ca3862478abb29e23f

drivers/mtd/spi/sf_probe.c

diff --git a/drivers/mtd/spi/sf_probe.c b/drivers/mtd/spi/sf_probe.c
index 00f8558..761833f 100644
--- a/drivers/mtd/spi/sf_probe.c
+++ b/drivers/mtd/spi/sf_probe.c
@@ -16,6 +16,10 @@
 
 #include "sf_internal.h"
 
+#ifdef CONFIG_AML_STORAGE
+extern int spi_flash_fit_storage(struct spi_flash *flash);
+#endif
+
 /**
  * spi_flash_probe_slave() - Probe for a SPI flash device on a bus
  *
@@ -45,8 +49,13 @@
 		goto err_read_id;
 
 #ifdef CONFIG_SPI_FLASH_MTD
-	ret = spi_flash_mtd_register(flash);
-#endif
+	/* for advanced support */
+	//ret = spi_flash_mtd_register(flash);
+#ifdef CONFIG_AML_STORAGE
+	if (!ret)
+		ret = spi_flash_fit_storage(flash);
+#endif /* CONFIG_AML_STORAGE */
+#endif /* CONFIG_SPI_FLASH_MTD */
 
 err_read_id:
 	spi_release_bus(spi);