6 files changed, 15 insertions, 10 deletions

diff --git a/block/blk-mq.c b/block/blk-mq.c
index 5189cb1e478a..702df07b980d 100644
--- a/block/blk-mq.c
+++ b/block/blk-mq.c
@@ -1776,7 +1776,8 @@ struct request_queue *blk_mq_init_queue(struct blk_mq_tag_set *set)
         if (!q)
                 goto err_hctxs;
 
-        if (percpu_ref_init(&q->mq_usage_counter, blk_mq_usage_counter_release))
+        if (percpu_ref_init(&q->mq_usage_counter, blk_mq_usage_counter_release,
+                            GFP_KERNEL))
                 goto err_map;
 
         setup_timer(&q->timeout, blk_mq_rq_timer, (unsigned long) q);
diff --git a/drivers/target/target_core_tpg.c b/drivers/target/target_core_tpg.c
index fddfae61222f..4ab6da338585 100644
--- a/drivers/target/target_core_tpg.c
+++ b/drivers/target/target_core_tpg.c
@@ -819,7 +819,8 @@ int core_tpg_add_lun(
 {
         int ret;
 
-        ret = percpu_ref_init(&lun->lun_ref, core_tpg_lun_ref_release);
+        ret = percpu_ref_init(&lun->lun_ref, core_tpg_lun_ref_release,
+                              GFP_KERNEL);
         if (ret < 0)
                 return ret;
 
diff --git a/fs/aio.c b/fs/aio.c
index bd7ec2cc2674..93fbcc0f5696 100644
--- a/fs/aio.c
+++ b/fs/aio.c
@@ -666,10 +666,10 @@ static struct kioctx *ioctx_alloc(unsigned nr_events)
 
         INIT_LIST_HEAD(&ctx->active_reqs);
 
-        if (percpu_ref_init(&ctx->users, free_ioctx_users))
+        if (percpu_ref_init(&ctx->users, free_ioctx_users, GFP_KERNEL))
                 goto err;
 
-        if (percpu_ref_init(&ctx->reqs, free_ioctx_reqs))
+        if (percpu_ref_init(&ctx->reqs, free_ioctx_reqs, GFP_KERNEL))
                 goto err;
 
         ctx->cpu = alloc_percpu(struct kioctx_cpu);
diff --git a/include/linux/percpu-refcount.h b/include/linux/percpu-refcount.h
index 3dfbf237cd8f..ee8325122dbd 100644
--- a/include/linux/percpu-refcount.h
+++ b/include/linux/percpu-refcount.h
@@ -49,6 +49,7 @@
 #include <linux/kernel.h>
 #include <linux/percpu.h>
 #include <linux/rcupdate.h>
+#include <linux/gfp.h>
 
 struct percpu_ref;
 typedef void (percpu_ref_func_t)(struct percpu_ref *);
@@ -66,7 +67,7 @@ struct percpu_ref {
 };
 
 int __must_check percpu_ref_init(struct percpu_ref *ref,
-                                 percpu_ref_func_t *release);
+                                 percpu_ref_func_t *release, gfp_t gfp);
 void percpu_ref_reinit(struct percpu_ref *ref);
 void percpu_ref_exit(struct percpu_ref *ref);
 void percpu_ref_kill_and_confirm(struct percpu_ref *ref,
diff --git a/kernel/cgroup.c b/kernel/cgroup.c
index 7dc8788cfd52..589b4d89a0a5 100644
--- a/kernel/cgroup.c
+++ b/kernel/cgroup.c
@@ -1628,7 +1628,7 @@ static int cgroup_setup_root(struct cgroup_root *root, unsigned int ss_mask)
                 goto out;
         root_cgrp->id = ret;
 
-        ret = percpu_ref_init(&root_cgrp->self.refcnt, css_release);
+        ret = percpu_ref_init(&root_cgrp->self.refcnt, css_release, GFP_KERNEL);
         if (ret)
                 goto out;
 
@@ -4487,7 +4487,7 @@ static int create_css(struct cgroup *cgrp, struct cgroup_subsys *ss,
 
         init_and_link_css(css, ss, cgrp);
 
-        err = percpu_ref_init(&css->refcnt, css_release);
+        err = percpu_ref_init(&css->refcnt, css_release, GFP_KERNEL);
         if (err)
                 goto err_free_css;
 
@@ -4555,7 +4555,7 @@ static int cgroup_mkdir(struct kernfs_node *parent_kn, const char *name,
                 goto out_unlock;
         }
 
-        ret = percpu_ref_init(&cgrp->self.refcnt, css_release);
+        ret = percpu_ref_init(&cgrp->self.refcnt, css_release, GFP_KERNEL);
         if (ret)
                 goto out_free_cgrp;
 
diff --git a/lib/percpu-refcount.c b/lib/percpu-refcount.c
index fe5a3342e960..ff9903264a91 100644
--- a/lib/percpu-refcount.c
+++ b/lib/percpu-refcount.c
@@ -40,6 +40,7 @@ static unsigned __percpu *pcpu_count_ptr(struct percpu_ref *ref)
  * percpu_ref_init - initialize a percpu refcount
  * @ref: percpu_ref to initialize
  * @release: function which will be called when refcount hits 0
+ * @gfp: allocation mask to use
  *
  * Initializes the refcount in single atomic counter mode with a refcount of 1;
  * analagous to atomic_set(ref, 1).
@@ -47,11 +48,12 @@ static unsigned __percpu *pcpu_count_ptr(struct percpu_ref *ref)
  * Note that @release must not sleep - it may potentially be called from RCU
  * callback context by percpu_ref_kill().
  */
-int percpu_ref_init(struct percpu_ref *ref, percpu_ref_func_t *release)
+int percpu_ref_init(struct percpu_ref *ref, percpu_ref_func_t *release,
+                    gfp_t gfp)
 {
         atomic_set(&ref->count, 1 + PCPU_COUNT_BIAS);
 
-        ref->pcpu_count_ptr = (unsigned long)alloc_percpu(unsigned);
+        ref->pcpu_count_ptr = (unsigned long)alloc_percpu_gfp(unsigned, gfp);
         if (!ref->pcpu_count_ptr)
                 return -ENOMEM;