kfree(cpd_to_bfqgd(cpd));
 }
 
-static struct blkg_policy_data *bfq_pd_alloc(gfp_t gfp, struct request_queue *q,
-                                            struct blkcg *blkcg)
+static struct blkg_policy_data *bfq_pd_alloc(struct gendisk *disk,
+               struct blkcg *blkcg, gfp_t gfp)
 {
        struct bfq_group *bfqg;
 
-       bfqg = kzalloc_node(sizeof(*bfqg), gfp, q->node);
+       bfqg = kzalloc_node(sizeof(*bfqg), gfp, disk->node_id);
        if (!bfqg)
                return NULL;
 
 
                        continue;
 
                /* alloc per-policy data and attach it to blkg */
-               pd = pol->pd_alloc_fn(gfp_mask, disk->queue, blkcg);
+               pd = pol->pd_alloc_fn(disk, blkcg, gfp_mask);
                if (!pd)
                        goto out_free_pds;
                blkg->pd[i] = pd;
                        pd = pd_prealloc;
                        pd_prealloc = NULL;
                } else {
-                       pd = pol->pd_alloc_fn(GFP_NOWAIT | __GFP_NOWARN, q,
-                                             blkg->blkcg);
+                       pd = pol->pd_alloc_fn(disk, blkg->blkcg,
+                                             GFP_NOWAIT | __GFP_NOWARN);
                }
 
                if (!pd) {
 
                        if (pd_prealloc)
                                pol->pd_free_fn(pd_prealloc);
-                       pd_prealloc = pol->pd_alloc_fn(GFP_KERNEL, q,
-                                                      blkg->blkcg);
+                       pd_prealloc = pol->pd_alloc_fn(disk, blkg->blkcg,
+                                                      GFP_KERNEL);
                        if (pd_prealloc)
                                goto retry;
                        else
 
 typedef void (blkcg_pol_init_cpd_fn)(struct blkcg_policy_data *cpd);
 typedef void (blkcg_pol_free_cpd_fn)(struct blkcg_policy_data *cpd);
 typedef void (blkcg_pol_bind_cpd_fn)(struct blkcg_policy_data *cpd);
-typedef struct blkg_policy_data *(blkcg_pol_alloc_pd_fn)(gfp_t gfp,
-                               struct request_queue *q, struct blkcg *blkcg);
+typedef struct blkg_policy_data *(blkcg_pol_alloc_pd_fn)(struct gendisk *disk,
+               struct blkcg *blkcg, gfp_t gfp);
 typedef void (blkcg_pol_init_pd_fn)(struct blkg_policy_data *pd);
 typedef void (blkcg_pol_online_pd_fn)(struct blkg_policy_data *pd);
 typedef void (blkcg_pol_offline_pd_fn)(struct blkg_policy_data *pd);
 
        kfree(container_of(cpd, struct ioc_cgrp, cpd));
 }
 
-static struct blkg_policy_data *ioc_pd_alloc(gfp_t gfp, struct request_queue *q,
-                                            struct blkcg *blkcg)
+static struct blkg_policy_data *ioc_pd_alloc(struct gendisk *disk,
+               struct blkcg *blkcg, gfp_t gfp)
 {
        int levels = blkcg->css.cgroup->level + 1;
        struct ioc_gq *iocg;
 
-       iocg = kzalloc_node(struct_size(iocg, ancestors, levels), gfp, q->node);
+       iocg = kzalloc_node(struct_size(iocg, ancestors, levels), gfp,
+                           disk->node_id);
        if (!iocg)
                return NULL;
 
 
                        iolat->max_depth, avg_lat, cur_win);
 }
 
-static struct blkg_policy_data *iolatency_pd_alloc(gfp_t gfp,
-                                                  struct request_queue *q,
-                                                  struct blkcg *blkcg)
+static struct blkg_policy_data *iolatency_pd_alloc(struct gendisk *disk,
+               struct blkcg *blkcg, gfp_t gfp)
 {
        struct iolatency_grp *iolat;
 
-       iolat = kzalloc_node(sizeof(*iolat), gfp, q->node);
+       iolat = kzalloc_node(sizeof(*iolat), gfp, disk->node_id);
        if (!iolat)
                return NULL;
        iolat->stats = __alloc_percpu_gfp(sizeof(struct latency_stat),
 
 }
 
 static struct blkg_policy_data *
-ioprio_alloc_pd(gfp_t gfp, struct request_queue *q, struct blkcg *blkcg)
+ioprio_alloc_pd(struct gendisk *disk, struct blkcg *blkcg, gfp_t gfp)
 {
        struct ioprio_blkg *ioprio_blkg;
 
 
        timer_setup(&sq->pending_timer, throtl_pending_timer_fn, 0);
 }
 
-static struct blkg_policy_data *throtl_pd_alloc(gfp_t gfp,
-                                               struct request_queue *q,
-                                               struct blkcg *blkcg)
+static struct blkg_policy_data *throtl_pd_alloc(struct gendisk *disk,
+               struct blkcg *blkcg, gfp_t gfp)
 {
        struct throtl_grp *tg;
        int rw;
 
-       tg = kzalloc_node(sizeof(*tg), gfp, q->node);
+       tg = kzalloc_node(sizeof(*tg), gfp, disk->node_id);
        if (!tg)
                return NULL;