diff options
Diffstat (limited to 'lib/closure.c')
-rw-r--r-- | lib/closure.c | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/lib/closure.c b/lib/closure.c index 2958169ce833..545b9726d1ea 100644 --- a/lib/closure.c +++ b/lib/closure.c @@ -22,6 +22,8 @@ static inline void closure_put_after_sub(struct closure *cl, int flags) panic("closure_put_after_sub: bogus flags %x remaining %i", flags, r); if (!r) { + smp_acquire__after_ctrl_dep(); + if (cl->fn && !(flags & CLOSURE_DESTRUCTOR)) { atomic_set(&cl->remaining, CLOSURE_REMAINING_INITIALIZER); @@ -44,7 +46,7 @@ static inline void closure_put_after_sub(struct closure *cl, int flags) /* For clearing flags with the same atomic op as a put */ void closure_sub(struct closure *cl, int v) { - closure_put_after_sub(cl, atomic_sub_return(v, &cl->remaining)); + closure_put_after_sub(cl, atomic_sub_return_release(v, &cl->remaining)); } EXPORT_SYMBOL(closure_sub); @@ -53,7 +55,7 @@ EXPORT_SYMBOL(closure_sub); */ void closure_put(struct closure *cl) { - closure_put_after_sub(cl, atomic_dec_return(&cl->remaining)); + closure_put_after_sub(cl, atomic_dec_return_release(&cl->remaining)); } EXPORT_SYMBOL(closure_put); |