diff --git a/drivers/of/address.c b/drivers/of/address.c
index 78a7dcbec7d8990ac37adad938a0aff3420423e2..65c3289fb2dc4d039aa827cd83145239f34ddb0c 100644
--- a/drivers/of/address.c
+++ b/drivers/of/address.c
@@ -712,7 +712,7 @@ int __weak pci_register_io_range(phys_addr_t addr, resource_size_t size)
 	}
 
 	/* add the range to the list */
-	range = kzalloc(sizeof(*range), GFP_KERNEL);
+	range = kzalloc(sizeof(*range), GFP_ATOMIC);
 	if (!range) {
 		err = -ENOMEM;
 		goto end_register;