1 files changed, 2 insertions, 2 deletions

diff --git a/arch/powerpc/lib/rheap.c b/arch/powerpc/lib/rheap.c
index 22c3b4f53de7..29b2941cada0 100644
--- a/arch/powerpc/lib/rheap.c
+++ b/arch/powerpc/lib/rheap.c
@@ -54,7 +54,7 @@ static int grow(rh_info_t * info, int max_blocks)
 
         new_blocks = max_blocks - info->max_blocks;
 
-        block = kmalloc(sizeof(rh_block_t) * max_blocks, GFP_KERNEL);
+        block = kmalloc(sizeof(rh_block_t) * max_blocks, GFP_ATOMIC);
         if (block == NULL)
                 return -ENOMEM;
 
@@ -258,7 +258,7 @@ rh_info_t *rh_create(unsigned int alignment)
         if ((alignment & (alignment - 1)) != 0)
                 return ERR_PTR(-EINVAL);
 
-        info = kmalloc(sizeof(*info), GFP_KERNEL);
+        info = kmalloc(sizeof(*info), GFP_ATOMIC);
         if (info == NULL)
                 return ERR_PTR(-ENOMEM);