2 files changed, 11 insertions, 2 deletions
diff --git a/src/soc/rockchip/rk3288/Makefile.inc b/src/soc/rockchip/rk3288/Makefile.inc
index 05003a210d..e0f32242a5 100644
--- a/src/soc/rockchip/rk3288/Makefile.inc
+++ b/src/soc/rockchip/rk3288/Makefile.inc
@@ -31,10 +31,10 @@ bootblock-y += spi.c
 romstage-y += cbmem.c
 romstage-y += timer.c
 romstage-y += monotonic_timer.c
-romstage-y += media.c
 romstage-$(CONFIG_DRIVERS_UART) += uart.c
 romstage-y += clock.c
 romstage-y += spi.c
+romstage-y += media.c
 
 ramstage-y += cbmem.c
 ramstage-y += timer.c
diff --git a/src/soc/rockchip/rk3288/media.c b/src/soc/rockchip/rk3288/media.c
index 75713eba7c..1389eb25fd 100644
--- a/src/soc/rockchip/rk3288/media.c
+++ b/src/soc/rockchip/rk3288/media.c
@@ -20,8 +20,17 @@
 #include <cbfs.h>
 #include <string.h>
 #include <console/console.h>
+#include "soc/rockchip/rk3288/spi.h"
 
 int init_default_cbfs_media(struct cbfs_media *media)
 {
-	return 0;
+#if defined(__BOOT_BLOCK__)
+	return initialize_rockchip_spi_cbfs_media(media,
+		(void *)CONFIG_CBFS_SRAM_CACHE_ADDRESS,
+		CONFIG_CBFS_SRAM_CACHE_SIZE);
+#else
+	return initialize_rockchip_spi_cbfs_media(media,
+		(void *)CONFIG_CBFS_DRAM_CACHE_ADDRESS,
+		CONFIG_CBFS_DRAM_CACHE_SIZE);
+#endif
 }