mutex_unlock(&blkcg_pol_mutex);
 
        for (i = 0; i < BLKCG_MAX_POLS; i++)
-               kfree(blkcg->pd[i]);
+               kfree(blkcg->cpd[i]);
        kfree(blkcg);
 }
 
                if (!pol || !pol->cpd_size)
                        continue;
 
-               BUG_ON(blkcg->pd[i]);
+               BUG_ON(blkcg->cpd[i]);
                cpd = kzalloc(pol->cpd_size, GFP_KERNEL);
                if (!cpd) {
                        ret = ERR_PTR(-ENOMEM);
                        goto free_pd_blkcg;
                }
-               blkcg->pd[i] = cpd;
+               blkcg->cpd[i] = cpd;
+               cpd->blkcg = blkcg;
                cpd->plid = i;
-               pol->cpd_init_fn(blkcg);
+               pol->cpd_init_fn(cpd);
        }
 
        spin_lock_init(&blkcg->lock);
 
 free_pd_blkcg:
        for (i--; i >= 0; i--)
-               kfree(blkcg->pd[i]);
+               kfree(blkcg->cpd[i]);
 free_blkcg:
        kfree(blkcg);
        mutex_unlock(&blkcg_pol_mutex);
                                goto err_free_cpds;
                        }
 
-                       blkcg->pd[pol->plid] = cpd;
+                       blkcg->cpd[pol->plid] = cpd;
+                       cpd->blkcg = blkcg;
                        cpd->plid = pol->plid;
-                       pol->cpd_init_fn(blkcg);
+                       pol->cpd_init_fn(cpd);
                }
        }
 
 err_free_cpds:
        if (pol->cpd_size) {
                list_for_each_entry(blkcg, &all_blkcgs, all_blkcgs_node) {
-                       kfree(blkcg->pd[pol->plid]);
-                       blkcg->pd[pol->plid] = NULL;
+                       kfree(blkcg->cpd[pol->plid]);
+                       blkcg->cpd[pol->plid] = NULL;
                }
        }
        blkcg_policy[pol->plid] = NULL;
 
        if (pol->cpd_size) {
                list_for_each_entry(blkcg, &all_blkcgs, all_blkcgs_node) {
-                       kfree(blkcg->pd[pol->plid]);
-                       blkcg->pd[pol->plid] = NULL;
+                       kfree(blkcg->cpd[pol->plid]);
+                       blkcg->cpd[pol->plid] = NULL;
                }
        }
        blkcg_policy[pol->plid] = NULL;
 
 /* Per-cgroup data */
 struct cfq_group_data {
        /* must be the first member */
-       struct blkcg_policy_data pd;
+       struct blkcg_policy_data cpd;
 
        unsigned int weight;
        unsigned int leaf_weight;
 static struct cfq_group_data
 *cpd_to_cfqgd(struct blkcg_policy_data *cpd)
 {
-       return cpd ? container_of(cpd, struct cfq_group_data, pd) : NULL;
+       return cpd ? container_of(cpd, struct cfq_group_data, cpd) : NULL;
 }
 
 static inline struct blkcg_gq *cfqg_to_blkg(struct cfq_group *cfqg)
 #endif
 }
 
-static void cfq_cpd_init(const struct blkcg *blkcg)
+static void cfq_cpd_init(struct blkcg_policy_data *cpd)
 {
-       struct cfq_group_data *cgd =
-               cpd_to_cfqgd(blkcg->pd[blkcg_policy_cfq.plid]);
+       struct cfq_group_data *cgd = cpd_to_cfqgd(cpd);
 
-       if (blkcg == &blkcg_root) {
+       if (cpd_to_blkcg(cpd) == &blkcg_root) {
                cgd->weight = 2 * CFQ_WEIGHT_DEFAULT;
                cgd->leaf_weight = 2 * CFQ_WEIGHT_DEFAULT;
        } else {
 
        struct blkcg_gq                 *blkg_hint;
        struct hlist_head               blkg_list;
 
-       struct blkcg_policy_data        *pd[BLKCG_MAX_POLS];
+       struct blkcg_policy_data        *cpd[BLKCG_MAX_POLS];
 
        struct list_head                all_blkcgs_node;
 #ifdef CONFIG_CGROUP_WRITEBACK
  * each policy handle per-blkcg data.
  */
 struct blkcg_policy_data {
-       /* the policy id this per-policy data belongs to */
+       /* the blkcg and policy id this per-policy data belongs to */
+       struct blkcg                    *blkcg;
        int                             plid;
 };
 
        struct rcu_head                 rcu_head;
 };
 
-typedef void (blkcg_pol_init_cpd_fn)(const struct blkcg *blkcg);
+typedef void (blkcg_pol_init_cpd_fn)(struct blkcg_policy_data *cpd);
 typedef struct blkg_policy_data *(blkcg_pol_alloc_pd_fn)(gfp_t gfp, int node);
 typedef void (blkcg_pol_init_pd_fn)(struct blkg_policy_data *pd);
 typedef void (blkcg_pol_online_pd_fn)(struct blkg_policy_data *pd);
 static inline struct blkcg_policy_data *blkcg_to_cpd(struct blkcg *blkcg,
                                                     struct blkcg_policy *pol)
 {
-       return blkcg ? blkcg->pd[pol->plid] : NULL;
+       return blkcg ? blkcg->cpd[pol->plid] : NULL;
 }
 
 /**
        return pd ? pd->blkg : NULL;
 }
 
+static inline struct blkcg *cpd_to_blkcg(struct blkcg_policy_data *cpd)
+{
+       return cpd ? cpd->blkcg : NULL;
+}
+
 /**
  * blkg_path - format cgroup path of blkg
  * @blkg: blkg of interest