#include <linux/delay.h>
 #include <linux/device.h>
 #include <linux/platform_device.h>
+#include <linux/platform_data/brcmnand.h>
 #include <linux/err.h>
 #include <linux/completion.h>
 #include <linux/interrupt.h>
        .attach_chip = brcmnand_attach_chip,
 };
 
-static int brcmnand_init_cs(struct brcmnand_host *host)
+static int brcmnand_init_cs(struct brcmnand_host *host,
+                           const char * const *part_probe_types)
 {
        struct brcmnand_controller *ctrl = host->ctrl;
        struct device *dev = ctrl->dev;
        if (ret)
                return ret;
 
-       ret = mtd_device_register(mtd, NULL, 0);
+       ret = mtd_device_parse_register(mtd, part_probe_types, NULL, NULL, 0);
        if (ret)
                nand_cleanup(chip);
 
 
 int brcmnand_probe(struct platform_device *pdev, struct brcmnand_soc *soc)
 {
+       struct brcmnand_platform_data *pd = dev_get_platdata(&pdev->dev);
        struct device *dev = &pdev->dev;
        struct device_node *dn = dev->of_node, *child;
        struct brcmnand_controller *ctrl;
+       struct brcmnand_host *host;
        struct resource *res;
        int ret;
 
-       /* We only support device-tree instantiation */
-       if (!dn)
-               return -ENODEV;
-
-       if (!of_match_node(brcmnand_of_match, dn))
+       if (dn && !of_match_node(brcmnand_of_match, dn))
                return -ENODEV;
 
        ctrl = devm_kzalloc(dev, sizeof(*ctrl), GFP_KERNEL);
        /* NAND register range */
        res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
        ctrl->nand_base = devm_ioremap_resource(dev, res);
-       if (IS_ERR(ctrl->nand_base))
+       if (IS_ERR(ctrl->nand_base) && !brcmnand_soc_has_ops(soc))
                return PTR_ERR(ctrl->nand_base);
 
        /* Enable clock before using NAND registers */
 
        for_each_available_child_of_node(dn, child) {
                if (of_device_is_compatible(child, "brcm,nandcs")) {
-                       struct brcmnand_host *host;
 
                        host = devm_kzalloc(dev, sizeof(*host), GFP_KERNEL);
                        if (!host) {
 
                        nand_set_flash_node(&host->chip, child);
 
-                       ret = brcmnand_init_cs(host);
+                       ret = brcmnand_init_cs(host, NULL);
                        if (ret) {
                                devm_kfree(dev, host);
                                continue; /* Try all chip-selects */
                }
        }
 
+       if (!list_empty(&ctrl->host_list))
+               return 0;
+
+       if (!pd) {
+               ret = -ENODEV;
+               goto err;
+       }
+
+       /* If we got there we must have been probing via platform data */
+       host = devm_kzalloc(dev, sizeof(*host), GFP_KERNEL);
+       if (!host) {
+               ret = -ENOMEM;
+               goto err;
+       }
+       host->pdev = pdev;
+       host->ctrl = ctrl;
+       host->cs = pd->chip_select;
+       host->chip.ecc.size = pd->ecc_stepsize;
+       host->chip.ecc.strength = pd->ecc_strength;
+
+       ret = brcmnand_init_cs(host, pd->part_probe_types);
+       if (ret)
+               goto err;
+
+       list_add_tail(&host->node, &ctrl->host_list);
+
        /* No chip-selects could initialize properly */
        if (list_empty(&ctrl->host_list)) {
                ret = -ENODEV;