#include <linux/of_address.h>
 #include <linux/of_irq.h>
 #include <linux/platform_device.h>
+#include <linux/pm_wakeirq.h>
 
 #include "mtu3.h"
 #include "mtu3_dr.h"
        if (IS_ERR(ssusb->ippc_base))
                return PTR_ERR(ssusb->ippc_base);
 
+       ssusb->wakeup_irq = platform_get_irq_byname_optional(pdev, "wakeup");
+       if (ssusb->wakeup_irq == -EPROBE_DEFER)
+               return ssusb->wakeup_irq;
+
        ssusb->dr_mode = usb_get_dr_mode(dev);
        if (ssusb->dr_mode == USB_DR_MODE_UNKNOWN)
                ssusb->dr_mode = USB_DR_MODE_OTG;
        ssusb_debugfs_create_root(ssusb);
 
        /* enable power domain */
+       pm_runtime_set_active(dev);
+       pm_runtime_use_autosuspend(dev);
+       pm_runtime_set_autosuspend_delay(dev, 4000);
        pm_runtime_enable(dev);
        pm_runtime_get_sync(dev);
-       device_enable_async_suspend(dev);
 
        ret = ssusb_rscs_init(ssusb);
        if (ret)
                goto comm_init_err;
 
+       if (ssusb->wakeup_irq > 0) {
+               ret = dev_pm_set_dedicated_wake_irq(dev, ssusb->wakeup_irq);
+               if (ret) {
+                       dev_err(dev, "failed to set wakeup irq %d\n", ssusb->wakeup_irq);
+                       goto comm_exit;
+               }
+               dev_info(dev, "wakeup irq %d\n", ssusb->wakeup_irq);
+       }
+
        ssusb_ip_sw_reset(ssusb);
 
        if (IS_ENABLED(CONFIG_USB_MTU3_HOST))
                goto comm_exit;
        }
 
+       device_enable_async_suspend(dev);
+       pm_runtime_mark_last_busy(dev);
+       pm_runtime_put_autosuspend(dev);
+       pm_runtime_forbid(dev);
+
        return 0;
 
 host_exit:
 comm_exit:
        ssusb_rscs_exit(ssusb);
 comm_init_err:
-       pm_runtime_put_sync(dev);
+       pm_runtime_put_noidle(dev);
        pm_runtime_disable(dev);
        ssusb_debugfs_remove_root(ssusb);
 
 {
        struct ssusb_mtk *ssusb = platform_get_drvdata(pdev);
 
+       pm_runtime_get_sync(&pdev->dev);
+
        switch (ssusb->dr_mode) {
        case USB_DR_MODE_PERIPHERAL:
                ssusb_gadget_exit(ssusb);
        }
 
        ssusb_rscs_exit(ssusb);
-       pm_runtime_put_sync(&pdev->dev);
-       pm_runtime_disable(&pdev->dev);
        ssusb_debugfs_remove_root(ssusb);
+       pm_runtime_disable(&pdev->dev);
+       pm_runtime_put_noidle(&pdev->dev);
+       pm_runtime_set_suspended(&pdev->dev);
 
        return 0;
 }
  * when support dual-role mode, we reject suspend when
  * it works as device mode;
  */
-static int __maybe_unused mtu3_suspend(struct device *dev)
+static int mtu3_suspend_common(struct device *dev, pm_message_t msg)
 {
        struct ssusb_mtk *ssusb = dev_get_drvdata(dev);
 
        return 0;
 }
 
-static int __maybe_unused mtu3_resume(struct device *dev)
+static int mtu3_resume_common(struct device *dev, pm_message_t msg)
 {
        struct ssusb_mtk *ssusb = dev_get_drvdata(dev);
        int ret;
        return ret;
 }
 
+static int __maybe_unused mtu3_suspend(struct device *dev)
+{
+       return mtu3_suspend_common(dev, PMSG_SUSPEND);
+}
+
+static int __maybe_unused mtu3_resume(struct device *dev)
+{
+       return mtu3_resume_common(dev, PMSG_SUSPEND);
+}
+
+static int __maybe_unused mtu3_runtime_suspend(struct device *dev)
+{
+       if (!device_may_wakeup(dev))
+               return 0;
+
+       return mtu3_suspend_common(dev, PMSG_AUTO_SUSPEND);
+}
+
+static int __maybe_unused mtu3_runtime_resume(struct device *dev)
+{
+       if (!device_may_wakeup(dev))
+               return 0;
+
+       return mtu3_resume_common(dev, PMSG_AUTO_SUSPEND);
+}
+
 static const struct dev_pm_ops mtu3_pm_ops = {
        SET_SYSTEM_SLEEP_PM_OPS(mtu3_suspend, mtu3_resume)
+       SET_RUNTIME_PM_OPS(mtu3_runtime_suspend,
+                          mtu3_runtime_resume, NULL)
 };
 
 #define DEV_PM_OPS (IS_ENABLED(CONFIG_PM) ? &mtu3_pm_ops : NULL)