1 files changed, 7 insertions, 3 deletions

diff --git a/mm/page_alloc.c b/mm/page_alloc.c
index 41c6e3aa059f..e25b6b24f844 100644
--- a/mm/page_alloc.c
+++ b/mm/page_alloc.c
@@ -2932,10 +2932,14 @@ void __init free_bootmem_with_active_regions(int nid,
 void __init work_with_active_regions(int nid, work_fn_t work_fn, void *data)
 {
         int i;
+        int ret;
 
-        for_each_active_range_index_in_nid(i, nid)
-                work_fn(early_node_map[i].start_pfn, early_node_map[i].end_pfn,
-                        data);
+        for_each_active_range_index_in_nid(i, nid) {
+                ret = work_fn(early_node_map[i].start_pfn,
+                              early_node_map[i].end_pfn, data);
+                if (ret)
+                        break;
+        }
 }
 /**
  * sparse_memory_present_with_active_regions - Call memory_present for each active range