Commit 3a587f47 authored by Peter Zijlstra's avatar Peter Zijlstra Committed by Linus Torvalds

lib: percpu_counter_set

Provide a method to set a percpu counter to a specified value.
Signed-off-by: default avatarPeter Zijlstra <a.p.zijlstra@chello.nl>
Signed-off-by: default avatarAndrew Morton <akpm@linux-foundation.org>
Signed-off-by: default avatarLinus Torvalds <torvalds@linux-foundation.org>
parent 20e89767
...@@ -32,6 +32,7 @@ struct percpu_counter { ...@@ -32,6 +32,7 @@ struct percpu_counter {
void percpu_counter_init(struct percpu_counter *fbc, s64 amount); void percpu_counter_init(struct percpu_counter *fbc, s64 amount);
void percpu_counter_destroy(struct percpu_counter *fbc); void percpu_counter_destroy(struct percpu_counter *fbc);
void percpu_counter_set(struct percpu_counter *fbc, s64 amount);
void __percpu_counter_add(struct percpu_counter *fbc, s64 amount, s32 batch); void __percpu_counter_add(struct percpu_counter *fbc, s64 amount, s32 batch);
s64 percpu_counter_sum(struct percpu_counter *fbc); s64 percpu_counter_sum(struct percpu_counter *fbc);
...@@ -75,6 +76,11 @@ static inline void percpu_counter_destroy(struct percpu_counter *fbc) ...@@ -75,6 +76,11 @@ static inline void percpu_counter_destroy(struct percpu_counter *fbc)
{ {
} }
static inline void percpu_counter_set(struct percpu_counter *fbc, s64 amount)
{
fbc->count = amount;
}
#define __percpu_counter_add(fbc, amount, batch) \ #define __percpu_counter_add(fbc, amount, batch) \
percpu_counter_add(fbc, amount) percpu_counter_add(fbc, amount)
......
...@@ -14,6 +14,20 @@ static LIST_HEAD(percpu_counters); ...@@ -14,6 +14,20 @@ static LIST_HEAD(percpu_counters);
static DEFINE_MUTEX(percpu_counters_lock); static DEFINE_MUTEX(percpu_counters_lock);
#endif #endif
void percpu_counter_set(struct percpu_counter *fbc, s64 amount)
{
int cpu;
spin_lock(&fbc->lock);
for_each_possible_cpu(cpu) {
s32 *pcount = per_cpu_ptr(fbc->counters, cpu);
*pcount = 0;
}
fbc->count = amount;
spin_unlock(&fbc->lock);
}
EXPORT_SYMBOL(percpu_counter_set);
void __percpu_counter_add(struct percpu_counter *fbc, s64 amount, s32 batch) void __percpu_counter_add(struct percpu_counter *fbc, s64 amount, s32 batch)
{ {
s64 count; s64 count;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment