1 files changed, 3 insertions, 10 deletions

diff --git a/drivers/of/of_reserved_mem.c b/drivers/of/of_reserved_mem.c
index 8c07d7da5256..6a36bc0b3d64 100644
--- a/drivers/of/of_reserved_mem.c
+++ b/drivers/of/of_reserved_mem.c
@@ -34,22 +34,15 @@ static int __init early_init_dt_alloc_reserved_memory_arch(phys_addr_t size,
 
         end = !end ? MEMBLOCK_ALLOC_ANYWHERE : end;
         align = !align ? SMP_CACHE_BYTES : align;
-        base = memblock_phys_alloc_range(size, align, 0, end);
+        base = memblock_find_in_range(start, end, size, align);
         if (!base)
                 return -ENOMEM;
 
-        /*
-         * Check if the allocated region fits in to start..end window
-         */
-        if (base < start) {
-                memblock_free(base, size);
-                return -ENOMEM;
-        }
-
         *res_base = base;
         if (nomap)
                 return memblock_remove(base, size);
-        return 0;
+
+        return memblock_reserve(base, size);
 }
 
 /**