WARN_ON(!base);
        ccm_base = base;
 
-       /* Reuse imx6q pm code */
-       imx6q_pm_set_ccm_base(base);
-
        /*                                              name                reg       shift width parent_names     num_parents */
        clks[IMX6SL_CLK_STEP]             = imx_clk_mux("step",             base + 0xc,  8,  1, step_sels,         ARRAY_SIZE(step_sels));
        clks[IMX6SL_CLK_PLL1_SW]          = imx_clk_mux("pll1_sw",          base + 0xc,  2,  1, pll1_sw_sels,      ARRAY_SIZE(pll1_sw_sels));
 
 
 struct imx6_pm_socdata {
        u32 ddr_type;
+       const char *ccm_compat;
        const char *mmdc_compat;
        const char *src_compat;
        const char *iomuxc_compat;
 };
 
 static const struct imx6_pm_socdata imx6q_pm_data __initconst = {
+       .ccm_compat = "fsl,imx6q-ccm",
        .mmdc_compat = "fsl,imx6q-mmdc",
        .src_compat = "fsl,imx6q-src",
        .iomuxc_compat = "fsl,imx6q-iomuxc",
 };
 
 static const struct imx6_pm_socdata imx6dl_pm_data __initconst = {
+       .ccm_compat = "fsl,imx6q-ccm",
        .mmdc_compat = "fsl,imx6q-mmdc",
        .src_compat = "fsl,imx6q-src",
        .iomuxc_compat = "fsl,imx6dl-iomuxc",
 };
 
 static const struct imx6_pm_socdata imx6sl_pm_data __initconst = {
+       .ccm_compat = "fsl,imx6sl-ccm",
        .mmdc_compat = "fsl,imx6sl-mmdc",
        .src_compat = "fsl,imx6sl-src",
        .iomuxc_compat = "fsl,imx6sl-iomuxc",
 };
 
 static const struct imx6_pm_socdata imx6sx_pm_data __initconst = {
+       .ccm_compat = "fsl,imx6sx-ccm",
        .mmdc_compat = "fsl,imx6sx-mmdc",
        .src_compat = "fsl,imx6sx-src",
        .iomuxc_compat = "fsl,imx6sx-iomuxc",
        .valid = imx6q_pm_valid,
 };
 
-void __init imx6q_pm_set_ccm_base(void __iomem *base)
-{
-       ccm_base = base;
-}
-
 static int __init imx6_pm_get_base(struct imx6_pm_base *base,
                                const char *compat)
 {
 
        /*
         * ccm physical address is not used by asm code currently,
-        * so get ccm virtual address directly, as we already have
-        * it from ccm driver.
+        * so get ccm virtual address directly.
         */
        pm_info->ccm_base.vbase = ccm_base;
 
 static void __init imx6_pm_common_init(const struct imx6_pm_socdata
                                        *socdata)
 {
+       struct device_node *np;
        struct regmap *gpr;
        int ret;
 
+       np = of_find_compatible_node(NULL, NULL, socdata->ccm_compat);
+       ccm_base = of_iomap(np, 0);
        WARN_ON(!ccm_base);
 
        imx6_set_lpm(WAIT_CLOCKED);