}
 #endif
 
+static int e820_register_one(struct resource *res, void *data)
+{
+       struct nd_region_desc ndr_desc;
+       struct nvdimm_bus *nvdimm_bus = data;
+
+       memset(&ndr_desc, 0, sizeof(ndr_desc));
+       ndr_desc.res = res;
+       ndr_desc.attr_groups = e820_pmem_region_attribute_groups;
+       ndr_desc.numa_node = e820_range_to_nid(res->start);
+       set_bit(ND_REGION_PAGEMAP, &ndr_desc.flags);
+       if (!nvdimm_pmem_region_create(nvdimm_bus, &ndr_desc))
+               return -ENXIO;
+       return 0;
+}
+
 static int e820_pmem_probe(struct platform_device *pdev)
 {
        static struct nvdimm_bus_descriptor nd_desc;
        struct device *dev = &pdev->dev;
        struct nvdimm_bus *nvdimm_bus;
-       struct resource *p;
+       int rc = -ENXIO;
 
        nd_desc.attr_groups = e820_pmem_attribute_groups;
        nd_desc.provider_name = "e820";
                goto err;
        platform_set_drvdata(pdev, nvdimm_bus);
 
-       for (p = iomem_resource.child; p ; p = p->sibling) {
-               struct nd_region_desc ndr_desc;
-
-               if (p->desc != IORES_DESC_PERSISTENT_MEMORY_LEGACY)
-                       continue;
-
-               memset(&ndr_desc, 0, sizeof(ndr_desc));
-               ndr_desc.res = p;
-               ndr_desc.attr_groups = e820_pmem_region_attribute_groups;
-               ndr_desc.numa_node = e820_range_to_nid(p->start);
-               set_bit(ND_REGION_PAGEMAP, &ndr_desc.flags);
-               if (!nvdimm_pmem_region_create(nvdimm_bus, &ndr_desc))
-                       goto err;
-       }
-
+       rc = walk_iomem_res_desc(IORES_DESC_PERSISTENT_MEMORY_LEGACY,
+                       IORESOURCE_MEM, 0, -1, nvdimm_bus, e820_register_one);
+       if (rc)
+               goto err;
        return 0;
-
- err:
+err:
        nvdimm_bus_unregister(nvdimm_bus);
        dev_err(dev, "failed to register legacy persistent memory ranges\n");
-       return -ENXIO;
+       return rc;
 }
 
 static struct platform_driver e820_pmem_driver = {