Add percpu_counter_mod64() to allow large modifications. Signed-off-by: Peter Zijlstra --- include/linux/percpu_counter.h | 17 +++++++++++++++++ lib/percpu_counter.c | 20 ++++++++++++++++++++ 2 files changed, 37 insertions(+) Index: linux-2.6/include/linux/percpu_counter.h =================================================================== --- linux-2.6.orig/include/linux/percpu_counter.h 2007-05-23 20:36:06.000000000 +0200 +++ linux-2.6/include/linux/percpu_counter.h 2007-05-23 20:37:41.000000000 +0200 @@ -33,6 +33,7 @@ struct percpu_counter { void percpu_counter_init(struct percpu_counter *fbc, s64 amount); void percpu_counter_destroy(struct percpu_counter *fbc); void __percpu_counter_mod(struct percpu_counter *fbc, s32 amount, s32 batch); +void __percpu_counter_mod64(struct percpu_counter *fbc, s64 amount, s32 batch); s64 percpu_counter_sum(struct percpu_counter *fbc); static inline void percpu_counter_mod(struct percpu_counter *fbc, s32 amount) @@ -40,6 +41,11 @@ static inline void percpu_counter_mod(st __percpu_counter_mod(fbc, amount, FBC_BATCH); } +static inline void percpu_counter_mod64(struct percpu_counter *fbc, s64 amount) +{ + __percpu_counter_mod64(fbc, amount, FBC_BATCH); +} + static inline s64 percpu_counter_read(struct percpu_counter *fbc) { return fbc->count; @@ -86,6 +92,17 @@ percpu_counter_mod(struct percpu_counter preempt_enable(); } +#define __percpu_counter_mod64(fbc, amount, batch) \ + percpu_counter_mod64(fbc, amount) + +static inline void +percpu_counter_mod64(struct percpu_counter *fbc, s64 amount) +{ + preempt_disable(); + fbc->count += amount; + preempt_enable(); +} + static inline s64 percpu_counter_read(struct percpu_counter *fbc) { return fbc->count; Index: linux-2.6/lib/percpu_counter.c =================================================================== --- linux-2.6.orig/lib/percpu_counter.c 2007-05-23 20:36:21.000000000 +0200 +++ linux-2.6/lib/percpu_counter.c 2007-05-23 20:37:34.000000000 +0200 @@ -34,6 +34,26 @@ void __percpu_counter_mod(struct percpu_ } EXPORT_SYMBOL(__percpu_counter_mod); +void __percpu_counter_mod64(struct percpu_counter *fbc, s64 amount, s32 batch) +{ + s64 count; + s32 *pcount; + int cpu = get_cpu(); + + pcount = per_cpu_ptr(fbc->counters, cpu); + count = *pcount + amount; + if (count >= batch || count <= -batch) { + spin_lock(&fbc->lock); + fbc->count += count; + *pcount = 0; + spin_unlock(&fbc->lock); + } else { + *pcount = count; + } + put_cpu(); +} +EXPORT_SYMBOL(__percpu_counter_mod64); + /* * Add up all the per-cpu counts, return the result. This is a more accurate * but much slower version of percpu_counter_read_positive() -- - To unsubscribe from this list: send the line "unsubscribe linux-kernel" in the body of a message to majordomo@vger.kernel.org More majordomo info at http://vger.kernel.org/majordomo-info.html Please read the FAQ at http://www.tux.org/lkml/