1 --- a/drivers/mtd/nand/rb4xx_nand.c
2 +++ b/drivers/mtd/nand/rb4xx_nand.c
4 * by the Free Software Foundation.
7 +#include <linux/kernel.h>
8 +#include <linux/module.h>
9 #include <linux/init.h>
10 #include <linux/mtd/nand.h>
11 #include <linux/mtd/mtd.h>
13 #include <linux/gpio.h>
14 #include <linux/slab.h>
16 -#include <asm/mach-ar71xx/ar71xx.h>
17 -#include <asm/mach-ar71xx/rb4xx_cpld.h>
18 +#include <asm/mach-ath79/ath79.h>
19 +#include <asm/mach-ath79/rb4xx_cpld.h>
21 #define DRV_NAME "rb4xx-nand"
22 #define DRV_VERSION "0.2.0"
23 @@ -238,12 +240,8 @@ static int __devinit rb4xx_nand_probe(st
27 -#ifdef CONFIG_MTD_PARTITIONS
28 - ret = add_mtd_partitions(&info->mtd, rb4xx_nand_partitions,
29 + mtd_device_register(&info->mtd, rb4xx_nand_partitions,
30 ARRAY_SIZE(rb4xx_nand_partitions));
32 - ret = add_mtd_device(&info->mtd);
35 goto err_release_nand;