diff options
author | Peter Zijlstra <a.p.zijlstra@chello.nl> | 2007-10-17 02:25:44 -0400 |
---|---|---|
committer | Linus Torvalds <torvalds@woody.linux-foundation.org> | 2007-10-17 11:42:44 -0400 |
commit | 3a587f47b82f96f19318c036e7b979fcd5c3848f (patch) | |
tree | 84b3254a7ea7524784f7e262d651cc332b3b0063 | |
parent | 20e89767096392a2cb2404437d3d181b8827af38 (diff) |
lib: percpu_counter_set
Provide a method to set a percpu counter to a specified value.
Signed-off-by: Peter Zijlstra <a.p.zijlstra@chello.nl>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
-rw-r--r-- | include/linux/percpu_counter.h | 6 | ||||
-rw-r--r-- | lib/percpu_counter.c | 14 |
2 files changed, 20 insertions, 0 deletions
diff --git a/include/linux/percpu_counter.h b/include/linux/percpu_counter.h index e8fa2da13ba6..2da92e8ce3dd 100644 --- a/include/linux/percpu_counter.h +++ b/include/linux/percpu_counter.h | |||
@@ -32,6 +32,7 @@ struct percpu_counter { | |||
32 | 32 | ||
33 | void percpu_counter_init(struct percpu_counter *fbc, s64 amount); | 33 | void percpu_counter_init(struct percpu_counter *fbc, s64 amount); |
34 | void percpu_counter_destroy(struct percpu_counter *fbc); | 34 | void percpu_counter_destroy(struct percpu_counter *fbc); |
35 | void percpu_counter_set(struct percpu_counter *fbc, s64 amount); | ||
35 | void __percpu_counter_add(struct percpu_counter *fbc, s64 amount, s32 batch); | 36 | void __percpu_counter_add(struct percpu_counter *fbc, s64 amount, s32 batch); |
36 | s64 percpu_counter_sum(struct percpu_counter *fbc); | 37 | s64 percpu_counter_sum(struct percpu_counter *fbc); |
37 | 38 | ||
@@ -75,6 +76,11 @@ static inline void percpu_counter_destroy(struct percpu_counter *fbc) | |||
75 | { | 76 | { |
76 | } | 77 | } |
77 | 78 | ||
79 | static inline void percpu_counter_set(struct percpu_counter *fbc, s64 amount) | ||
80 | { | ||
81 | fbc->count = amount; | ||
82 | } | ||
83 | |||
78 | #define __percpu_counter_add(fbc, amount, batch) \ | 84 | #define __percpu_counter_add(fbc, amount, batch) \ |
79 | percpu_counter_add(fbc, amount) | 85 | percpu_counter_add(fbc, amount) |
80 | 86 | ||
diff --git a/lib/percpu_counter.c b/lib/percpu_counter.c index f7ac68c4c375..c9708db9b8d3 100644 --- a/lib/percpu_counter.c +++ b/lib/percpu_counter.c | |||
@@ -14,6 +14,20 @@ static LIST_HEAD(percpu_counters); | |||
14 | static DEFINE_MUTEX(percpu_counters_lock); | 14 | static DEFINE_MUTEX(percpu_counters_lock); |
15 | #endif | 15 | #endif |
16 | 16 | ||
17 | void percpu_counter_set(struct percpu_counter *fbc, s64 amount) | ||
18 | { | ||
19 | int cpu; | ||
20 | |||
21 | spin_lock(&fbc->lock); | ||
22 | for_each_possible_cpu(cpu) { | ||
23 | s32 *pcount = per_cpu_ptr(fbc->counters, cpu); | ||
24 | *pcount = 0; | ||
25 | } | ||
26 | fbc->count = amount; | ||
27 | spin_unlock(&fbc->lock); | ||
28 | } | ||
29 | EXPORT_SYMBOL(percpu_counter_set); | ||
30 | |||
17 | void __percpu_counter_add(struct percpu_counter *fbc, s64 amount, s32 batch) | 31 | void __percpu_counter_add(struct percpu_counter *fbc, s64 amount, s32 batch) |
18 | { | 32 | { |
19 | s64 count; | 33 | s64 count; |