1 files changed, 9 insertions, 1 deletions

diff --git a/include/linux/percpu-refcount.h b/include/linux/percpu-refcount.h
index b297cd1cd4f1..7aef0abc194a 100644
--- a/include/linux/percpu-refcount.h
+++ b/include/linux/percpu-refcount.h
@@ -75,14 +75,21 @@ enum {
          * operation using percpu_ref_switch_to_percpu().  If initialized
          * with this flag, the ref will stay in atomic mode until
          * percpu_ref_switch_to_percpu() is invoked on it.
+         * Implies ALLOW_REINIT.
          */
         PERCPU_REF_INIT_ATOMIC  = 1 << 0,
 
         /*
          * Start dead w/ ref == 0 in atomic mode.  Must be revived with
-         * percpu_ref_reinit() before used.  Implies INIT_ATOMIC.
+         * percpu_ref_reinit() before used.  Implies INIT_ATOMIC and
+         * ALLOW_REINIT.
          */
         PERCPU_REF_INIT_DEAD    = 1 << 1,
+
+        /*
+         * Allow switching from atomic mode to percpu mode.
+         */
+        PERCPU_REF_ALLOW_REINIT = 1 << 2,
 };
 
 struct percpu_ref {
@@ -95,6 +102,7 @@ struct percpu_ref {
         percpu_ref_func_t       *release;
         percpu_ref_func_t       *confirm_switch;
         bool                    force_atomic:1;
+        bool                    allow_reinit:1;
         struct rcu_head         rcu;
 };