diff options
-rw-r--r-- | mm/mempool.c | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/mm/mempool.c b/mm/mempool.c index e3a802a0cea..d9049811f35 100644 --- a/mm/mempool.c +++ b/mm/mempool.c | |||
@@ -221,14 +221,23 @@ repeat_alloc: | |||
221 | return element; | 221 | return element; |
222 | } | 222 | } |
223 | 223 | ||
224 | /* We must not sleep in the GFP_ATOMIC case */ | 224 | /* |
225 | * We use gfp mask w/o __GFP_WAIT or IO for the first round. If | ||
226 | * alloc failed with that and @pool was empty, retry immediately. | ||
227 | */ | ||
228 | if (gfp_temp != gfp_mask) { | ||
229 | spin_unlock_irqrestore(&pool->lock, flags); | ||
230 | gfp_temp = gfp_mask; | ||
231 | goto repeat_alloc; | ||
232 | } | ||
233 | |||
234 | /* We must not sleep if !__GFP_WAIT */ | ||
225 | if (!(gfp_mask & __GFP_WAIT)) { | 235 | if (!(gfp_mask & __GFP_WAIT)) { |
226 | spin_unlock_irqrestore(&pool->lock, flags); | 236 | spin_unlock_irqrestore(&pool->lock, flags); |
227 | return NULL; | 237 | return NULL; |
228 | } | 238 | } |
229 | 239 | ||
230 | /* Let's wait for someone else to return an element to @pool */ | 240 | /* Let's wait for someone else to return an element to @pool */ |
231 | gfp_temp = gfp_mask; | ||
232 | init_wait(&wait); | 241 | init_wait(&wait); |
233 | prepare_to_wait(&pool->wait, &wait, TASK_UNINTERRUPTIBLE); | 242 | prepare_to_wait(&pool->wait, &wait, TASK_UNINTERRUPTIBLE); |
234 | 243 | ||