1 files changed, 2 insertions, 0 deletions

diff --git a/mm/dmapool.c b/mm/dmapool.c
index 3df063706f53..4df2de77e069 100644
--- a/mm/dmapool.c
+++ b/mm/dmapool.c
@@ -311,6 +311,8 @@ void *dma_pool_alloc(struct dma_pool *pool, gfp_t mem_flags,
         size_t offset;
         void *retval;
 
+        might_sleep_if(mem_flags & __GFP_WAIT);
+
         spin_lock_irqsave(&pool->lock, flags);
  restart:
         list_for_each_entry(page, &pool->page_list, page_list) {