Add complete_n() call.

2 files changed, 13 insertions, 1 deletions

diff --git a/include/linux/completion.h b/include/linux/completion.h
index 33d6aaf94447..5b55e972be2c 100644
--- a/include/linux/completion.h
+++ b/include/linux/completion.h
@@ -51,7 +51,7 @@ extern unsigned long wait_for_completion_interruptible_timeout(
 
 extern void complete(struct completion *);
 extern void complete_all(struct completion *);
-
+extern void complete_n(struct completion *, int n);
 #define INIT_COMPLETION(x)      ((x).done = 0)
 
 #endif
diff --git a/kernel/sched.c b/kernel/sched.c
index 09b529cf2bd8..c7d7d535288a 100644
--- a/kernel/sched.c
+++ b/kernel/sched.c
@@ -3923,6 +3923,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)
 {