1 files changed, 6 insertions, 1 deletions

diff --git a/mm/vmalloc.c b/mm/vmalloc.c
index 4ac776f10ad1..3308007d8427 100644
--- a/mm/vmalloc.c
+++ b/mm/vmalloc.c
@@ -365,7 +365,7 @@ static struct vmap_area *alloc_vmap_area(unsigned long size,
         BUG_ON(offset_in_page(size));
         BUG_ON(!is_power_of_2(align));
 
-        might_sleep_if(gfpflags_allow_blocking(gfp_mask));
+        might_sleep();
 
         va = kmalloc_node(sizeof(struct vmap_area),
                         gfp_mask & GFP_RECLAIM_MASK, node);
@@ -1037,6 +1037,8 @@ void vm_unmap_aliases(void)
         if (unlikely(!vmap_initialized))
                 return;
 
+        might_sleep();
+
         for_each_possible_cpu(cpu) {
                 struct vmap_block_queue *vbq = &per_cpu(vmap_block_queue, cpu);
                 struct vmap_block *vb;
@@ -1080,6 +1082,7 @@ void vm_unmap_ram(const void *mem, unsigned int count)
         unsigned long addr = (unsigned long)mem;
         struct vmap_area *va;
 
+        might_sleep();
         BUG_ON(!addr);
         BUG_ON(addr < VMALLOC_START);
         BUG_ON(addr > VMALLOC_END);
@@ -1431,6 +1434,8 @@ struct vm_struct *remove_vm_area(const void *addr)
 {
         struct vmap_area *va;
 
+        might_sleep();
+
         va = find_vmap_area((unsigned long)addr);
         if (va && va->flags & VM_VM_AREA) {
                 struct vm_struct *vm = va->vm;