am_phy->usb_phy_gen.phy.shutdown = am335x_shutdown;
 
        platform_set_drvdata(pdev, am_phy);
+       device_init_wakeup(dev, true);
+
+       /*
+        * If we leave PHY wakeup enabled then AM33XX wakes up
+        * immediately from DS0. To avoid this we mark dev->power.can_wakeup
+        * to false. The same is checked in suspend routine to decide
+        * on whether to enable PHY wakeup or not.
+        * PHY wakeup works fine in standby mode, there by allowing us to
+        * handle remote wakeup, wakeup on disconnect and connect.
+        */
+
+       device_set_wakeup_enable(dev, false);
+       phy_ctrl_power(am_phy->phy_ctrl, am_phy->id, false);
 
        return 0;
 }
        return 0;
 }
 
-#ifdef CONFIG_PM_RUNTIME
-
-static int am335x_phy_runtime_suspend(struct device *dev)
+#ifdef CONFIG_PM_SLEEP
+static int am335x_phy_suspend(struct device *dev)
 {
        struct platform_device  *pdev = to_platform_device(dev);
        struct am335x_phy *am_phy = platform_get_drvdata(pdev);
 
+       /*
+        * Enable phy wakeup only if dev->power.can_wakeup is true.
+        * Make sure to enable wakeup to support remote wakeup  in
+        * standby mode ( same is not supported in OFF(DS0) mode).
+        * Enable it by doing
+        * echo enabled > /sys/bus/platform/devices/<usb-phy-id>/power/wakeup
+        */
+
        if (device_may_wakeup(dev))
                phy_ctrl_wkup(am_phy->phy_ctrl, am_phy->id, true);
+
        phy_ctrl_power(am_phy->phy_ctrl, am_phy->id, false);
+
        return 0;
 }
 
-static int am335x_phy_runtime_resume(struct device *dev)
+static int am335x_phy_resume(struct device *dev)
 {
        struct platform_device  *pdev = to_platform_device(dev);
        struct am335x_phy       *am_phy = platform_get_drvdata(pdev);
 
        phy_ctrl_power(am_phy->phy_ctrl, am_phy->id, true);
+
        if (device_may_wakeup(dev))
                phy_ctrl_wkup(am_phy->phy_ctrl, am_phy->id, false);
+
        return 0;
 }
+#define DEV_PM_OPS     (&am335x_pm_ops)
+#else
+#define DEV_PM_OPS     NULL
+#endif
+
 
 static const struct dev_pm_ops am335x_pm_ops = {
-       SET_RUNTIME_PM_OPS(am335x_phy_runtime_suspend,
-                       am335x_phy_runtime_resume, NULL)
+       .suspend = am335x_phy_suspend,
+       .resume = am335x_phy_resume,
 };
 
-#define DEV_PM_OPS     (&am335x_pm_ops)
-#else
-#define DEV_PM_OPS     NULL
-#endif
-
 static const struct of_device_id am335x_phy_ids[] = {
        { .compatible = "ti,am335x-usb-phy" },
        { }