return ret < 0 ? ret : 0;
 }
 
-#ifdef CONFIG_MEMORY_HOTPLUG
 int htab_remove_mapping(unsigned long vstart, unsigned long vend,
                      int psize, int ssize)
 {
 
        return ret;
 }
-#endif /* CONFIG_MEMORY_HOTPLUG */
 
 static int __init htab_dt_scan_seg_sizes(unsigned long node,
                                         const char *uname, int depth,
 #ifdef CONFIG_MEMORY_HOTPLUG
 int create_section_mapping(unsigned long start, unsigned long end)
 {
-       return htab_bolt_mapping(start, end, __pa(start),
-                                pgprot_val(PAGE_KERNEL), mmu_linear_psize,
-                                mmu_kernel_ssize);
+       int rc = htab_bolt_mapping(start, end, __pa(start),
+                                  pgprot_val(PAGE_KERNEL), mmu_linear_psize,
+                                  mmu_kernel_ssize);
+
+       if (rc < 0) {
+               int rc2 = htab_remove_mapping(start, end, mmu_linear_psize,
+                                             mmu_kernel_ssize);
+               BUG_ON(rc2 && (rc2 != -ENOENT));
+       }
+       return rc;
 }
 
 int remove_section_mapping(unsigned long start, unsigned long end)
 
  */
 
 #ifdef CONFIG_PPC_BOOK3E
-static void __meminit vmemmap_create_mapping(unsigned long start,
-                                            unsigned long page_size,
-                                            unsigned long phys)
+static int __meminit vmemmap_create_mapping(unsigned long start,
+                                           unsigned long page_size,
+                                           unsigned long phys)
 {
        /* Create a PTE encoding without page size */
        unsigned long i, flags = _PAGE_PRESENT | _PAGE_ACCESSED |
         */
        for (i = 0; i < page_size; i += PAGE_SIZE)
                BUG_ON(map_kernel_page(start + i, phys, flags));
+
+       return 0;
 }
 
 #ifdef CONFIG_MEMORY_HOTPLUG
 }
 #endif
 #else /* CONFIG_PPC_BOOK3E */
-static void __meminit vmemmap_create_mapping(unsigned long start,
-                                            unsigned long page_size,
-                                            unsigned long phys)
+static int __meminit vmemmap_create_mapping(unsigned long start,
+                                           unsigned long page_size,
+                                           unsigned long phys)
 {
-       int  mapped = htab_bolt_mapping(start, start + page_size, phys,
-                                       pgprot_val(PAGE_KERNEL),
-                                       mmu_vmemmap_psize,
-                                       mmu_kernel_ssize);
-       BUG_ON(mapped < 0);
+       int rc = htab_bolt_mapping(start, start + page_size, phys,
+                                  pgprot_val(PAGE_KERNEL),
+                                  mmu_vmemmap_psize, mmu_kernel_ssize);
+       if (rc < 0) {
+               int rc2 = htab_remove_mapping(start, start + page_size,
+                                             mmu_vmemmap_psize,
+                                             mmu_kernel_ssize);
+               BUG_ON(rc2 && (rc2 != -ENOENT));
+       }
+       return rc;
 }
 
 #ifdef CONFIG_MEMORY_HOTPLUG
 
        for (; start < end; start += page_size) {
                void *p;
+               int rc;
 
                if (vmemmap_populated(start, page_size))
                        continue;
                pr_debug("      * %016lx..%016lx allocated at %p\n",
                         start, start + page_size, p);
 
-               vmemmap_create_mapping(start, page_size, __pa(p));
+               rc = vmemmap_create_mapping(start, page_size, __pa(p));
+               if (rc < 0) {
+                       pr_warning(
+                               "vmemmap_populate: Unable to create vmemmap mapping: %d\n",
+                               rc);
+                       return -EFAULT;
+               }
        }
 
        return 0;
 
        struct zone *zone;
        unsigned long start_pfn = start >> PAGE_SHIFT;
        unsigned long nr_pages = size >> PAGE_SHIFT;
+       int rc;
 
        pgdata = NODE_DATA(nid);
 
        start = (unsigned long)__va(start);
-       if (create_section_mapping(start, start + size))
-               return -EINVAL;
+       rc = create_section_mapping(start, start + size);
+       if (rc) {
+               pr_warning(
+                       "Unable to create mapping for hot added memory 0x%llx..0x%llx: %d\n",
+                       start, start + size, rc);
+               return -EFAULT;
+       }
 
        /* this should work for most non-highmem platforms */
        zone = pgdata->node_zones +