struct clk *clk;
        spinlock_t lock; /* protect access to IPCC registers */
        int irqs[IPCC_IRQ_NUM];
-       int wkp;
        u32 proc_id;
        u32 n_chans;
        u32 xcr;
 
        /* wakeup */
        if (of_property_read_bool(np, "wakeup-source")) {
-               ipcc->wkp = platform_get_irq_byname(pdev, "wakeup");
-               if (ipcc->wkp < 0) {
-                       if (ipcc->wkp != -EPROBE_DEFER)
-                               dev_err(dev, "could not get wakeup IRQ\n");
-                       ret = ipcc->wkp;
-                       goto err_clk;
-               }
-
                device_set_wakeup_capable(dev, true);
-               ret = dev_pm_set_dedicated_wake_irq(dev, ipcc->wkp);
+
+               ret = dev_pm_set_wake_irq(dev, ipcc->irqs[IPCC_IRQ_RX]);
                if (ret) {
                        dev_err(dev, "Failed to set wake up irq\n");
                        goto err_init_wkp;
        return 0;
 
 err_irq_wkp:
-       if (ipcc->wkp)
+       if (of_property_read_bool(np, "wakeup-source"))
                dev_pm_clear_wake_irq(dev);
 err_init_wkp:
-       device_init_wakeup(dev, false);
+       device_set_wakeup_capable(dev, false);
 err_clk:
        clk_disable_unprepare(ipcc->clk);
        return ret;
 
 static int stm32_ipcc_remove(struct platform_device *pdev)
 {
-       struct stm32_ipcc *ipcc = platform_get_drvdata(pdev);
+       struct device *dev = &pdev->dev;
 
-       if (ipcc->wkp)
+       if (of_property_read_bool(dev->of_node, "wakeup-source"))
                dev_pm_clear_wake_irq(&pdev->dev);
 
-       device_init_wakeup(&pdev->dev, false);
+       device_set_wakeup_capable(dev, false);
 
        return 0;
 }
 
 #ifdef CONFIG_PM_SLEEP
-static void stm32_ipcc_set_irq_wake(struct device *dev, bool enable)
-{
-       struct stm32_ipcc *ipcc = dev_get_drvdata(dev);
-       unsigned int i;
-
-       if (device_may_wakeup(dev))
-               for (i = 0; i < IPCC_IRQ_NUM; i++)
-                       irq_set_irq_wake(ipcc->irqs[i], enable);
-}
-
 static int stm32_ipcc_suspend(struct device *dev)
 {
        struct stm32_ipcc *ipcc = dev_get_drvdata(dev);
        ipcc->xmr = readl_relaxed(ipcc->reg_proc + IPCC_XMR);
        ipcc->xcr = readl_relaxed(ipcc->reg_proc + IPCC_XCR);
 
-       stm32_ipcc_set_irq_wake(dev, true);
-
        return 0;
 }
 
 {
        struct stm32_ipcc *ipcc = dev_get_drvdata(dev);
 
-       stm32_ipcc_set_irq_wake(dev, false);
-
        writel_relaxed(ipcc->xmr, ipcc->reg_proc + IPCC_XMR);
        writel_relaxed(ipcc->xcr, ipcc->reg_proc + IPCC_XCR);