sbitmap: Try each queue to wake up at least one waiter

commit 26edb30dd1 upstream.

Jan reported the new algorithm as merged might be problematic if the
queue being awaken becomes empty between the waitqueue_active inside
sbq_wake_ptr check and the wake up.  If that happens, wake_up_nr will
not wake up any waiter and we loose too many wake ups.  In order to
guarantee progress, we need to wake up at least one waiter here, if
there are any.  This now requires trying to wake up from every queue.

Instead of walking through all the queues with sbq_wake_ptr, this call
moves the wake up inside that function.  In a previous version of the
patch, I found that updating wake_index several times when walking
through queues had a measurable overhead.  This ensures we only update
it once, at the end.

Fixes: 4f8126bb23 ("sbitmap: Use single per-bitmap counting to wake up queued tags")
Reported-by: Jan Kara <jack@suse.cz>
Signed-off-by: Gabriel Krisman Bertazi <krisman@suse.de>
Reviewed-by: Jan Kara <jack@suse.cz>
Link: https://lore.kernel.org/r/20221115224553.23594-4-krisman@suse.de
Signed-off-by: Jens Axboe <axboe@kernel.dk>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>
This commit is contained in:
Gabriel Krisman Bertazi 2022-11-15 17:45:53 -05:00 committed by Greg Kroah-Hartman
parent d710b1e91b
commit ff9571a4de
1 changed files with 12 additions and 16 deletions

View File

@ -555,12 +555,12 @@ void sbitmap_queue_min_shallow_depth(struct sbitmap_queue *sbq,
} }
EXPORT_SYMBOL_GPL(sbitmap_queue_min_shallow_depth); EXPORT_SYMBOL_GPL(sbitmap_queue_min_shallow_depth);
static struct sbq_wait_state *sbq_wake_ptr(struct sbitmap_queue *sbq) static void __sbitmap_queue_wake_up(struct sbitmap_queue *sbq, int nr)
{ {
int i, wake_index; int i, wake_index;
if (!atomic_read(&sbq->ws_active)) if (!atomic_read(&sbq->ws_active))
return NULL; return;
wake_index = atomic_read(&sbq->wake_index); wake_index = atomic_read(&sbq->wake_index);
for (i = 0; i < SBQ_WAIT_QUEUES; i++) { for (i = 0; i < SBQ_WAIT_QUEUES; i++) {
@ -574,20 +574,22 @@ static struct sbq_wait_state *sbq_wake_ptr(struct sbitmap_queue *sbq)
*/ */
wake_index = sbq_index_inc(wake_index); wake_index = sbq_index_inc(wake_index);
if (waitqueue_active(&ws->wait)) { /*
if (wake_index != atomic_read(&sbq->wake_index)) * It is sufficient to wake up at least one waiter to
atomic_set(&sbq->wake_index, wake_index); * guarantee forward progress.
return ws; */
} if (waitqueue_active(&ws->wait) &&
wake_up_nr(&ws->wait, nr))
break;
} }
return NULL; if (wake_index != atomic_read(&sbq->wake_index))
atomic_set(&sbq->wake_index, wake_index);
} }
void sbitmap_queue_wake_up(struct sbitmap_queue *sbq, int nr) void sbitmap_queue_wake_up(struct sbitmap_queue *sbq, int nr)
{ {
unsigned int wake_batch = READ_ONCE(sbq->wake_batch); unsigned int wake_batch = READ_ONCE(sbq->wake_batch);
struct sbq_wait_state *ws = NULL;
unsigned int wakeups; unsigned int wakeups;
if (!atomic_read(&sbq->ws_active)) if (!atomic_read(&sbq->ws_active))
@ -599,16 +601,10 @@ void sbitmap_queue_wake_up(struct sbitmap_queue *sbq, int nr)
do { do {
if (atomic_read(&sbq->completion_cnt) - wakeups < wake_batch) if (atomic_read(&sbq->completion_cnt) - wakeups < wake_batch)
return; return;
if (!ws) {
ws = sbq_wake_ptr(sbq);
if (!ws)
return;
}
} while (!atomic_try_cmpxchg(&sbq->wakeup_cnt, } while (!atomic_try_cmpxchg(&sbq->wakeup_cnt,
&wakeups, wakeups + wake_batch)); &wakeups, wakeups + wake_batch));
wake_up_nr(&ws->wait, wake_batch); __sbitmap_queue_wake_up(sbq, wake_batch);
} }
EXPORT_SYMBOL_GPL(sbitmap_queue_wake_up); EXPORT_SYMBOL_GPL(sbitmap_queue_wake_up);