unsigned int cpt_num_devs;
        unsigned int max_pwm_cnt;
        unsigned int max_prescale;
+       struct sti_cpt_ddata *ddata;
 };
 
 struct sti_pwm_chip {
 {
        struct sti_pwm_chip *pc = to_sti_pwmchip(chip);
        struct sti_pwm_compat_data *cdata = pc->cdata;
-       struct sti_cpt_ddata *ddata = pwm_get_chip_data(pwm);
+       struct sti_cpt_ddata *ddata = &cdata->ddata[pwm->hwpwm];
        struct device *dev = pc->dev;
        unsigned int effective_ticks;
        unsigned long long high, low;
        while (cpt_int_stat) {
                devicenum = ffs(cpt_int_stat) - 1;
 
-               ddata = pwm_get_chip_data(&pc->chip.pwms[devicenum]);
+               ddata = &pc->cdata->ddata[devicenum];
 
                /*
                 * Capture input:
                        dev_err(dev, "failed to prepare clock\n");
                        return ret;
                }
+
+               cdata->ddata = devm_kzalloc(dev, cdata->cpt_num_devs * sizeof(*cdata->ddata), GFP_KERNEL);
+               if (!cdata->ddata)
+                       return -ENOMEM;
        }
 
        pc->chip.dev = dev;
        pc->chip.ops = &sti_pwm_ops;
        pc->chip.npwm = pc->cdata->pwm_num_devs;
 
-       ret = pwmchip_add(&pc->chip);
-       if (ret < 0) {
-               clk_unprepare(pc->pwm_clk);
-               clk_unprepare(pc->cpt_clk);
-               return ret;
-       }
-
        for (i = 0; i < cdata->cpt_num_devs; i++) {
-               struct sti_cpt_ddata *ddata;
-
-               ddata = devm_kzalloc(dev, sizeof(*ddata), GFP_KERNEL);
-               if (!ddata)
-                       return -ENOMEM;
+               struct sti_cpt_ddata *ddata = &cdata->ddata[i];
 
                init_waitqueue_head(&ddata->wait);
                mutex_init(&ddata->lock);
+       }
 
-               pwm_set_chip_data(&pc->chip.pwms[i], ddata);
+       ret = pwmchip_add(&pc->chip);
+       if (ret < 0) {
+               clk_unprepare(pc->pwm_clk);
+               clk_unprepare(pc->cpt_clk);
+               return ret;
        }
 
        platform_set_drvdata(pdev, pc);