add complete_n() call to the completion API.

This is usefule for releasing exactly n tasks without having to
call compete() repeatedly.

1 files changed, 12 insertions, 0 deletions

diff --git a/kernel/sched.c b/kernel/sched.c
index 4890a12786..5d31bc9d56 100644
--- a/kernel/sched.c
+++ b/kernel/sched.c
@@ -3895,6 +3895,18 @@ void complete_all(struct completion *x)
 }
 EXPORT_SYMBOL(complete_all);
 
+void complete_n(struct completion *x, int n)
+{
+        unsigned long flags;
+
+        spin_lock_irqsave(&x->wait.lock, flags);
+        x->done += n;
+        __wake_up_common(&x->wait, TASK_UNINTERRUPTIBLE | TASK_INTERRUPTIBLE,
+                         n, 0, NULL);
+        spin_unlock_irqrestore(&x->wait.lock, flags);
+}
+EXPORT_SYMBOL(complete_n);
+
 static inline long __sched
 do_wait_for_common(struct completion *x, long timeout, int state)
 {