1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/mtd/maps/uclinux.c b/drivers/mtd/maps/uclinux.c
index d4314fb88212..35009294b435 100644
--- a/drivers/mtd/maps/uclinux.c
+++ b/drivers/mtd/maps/uclinux.c
@@ -89,7 +89,11 @@ static int __init uclinux_mtd_init(void)
         mtd->priv = mapp;
 
         uclinux_ram_mtdinfo = mtd;
+#ifdef CONFIG_MTD_PARTITIONS
         add_mtd_partitions(mtd, uclinux_romfs, NUM_PARTITIONS);
+#else
+        add_mtd_device(mtd);
+#endif
 
         return(0);
 }
@@ -99,7 +103,11 @@ static int __init uclinux_mtd_init(void)
 static void __exit uclinux_mtd_cleanup(void)
 {
         if (uclinux_ram_mtdinfo) {
+#ifdef CONFIG_MTD_PARTITIONS
                 del_mtd_partitions(uclinux_ram_mtdinfo);
+#else
+                del_mtd_device(uclinux_ram_mtdinfo);
+#endif
                 map_destroy(uclinux_ram_mtdinfo);
                 uclinux_ram_mtdinfo = NULL;
         }