2 files changed, 14 insertions, 21 deletions

diff --git a/drivers/usb/host/ehci-omap.c b/drivers/usb/host/ehci-omap.c
index e39b0297bad1..568cefbd63a2 100644
--- a/drivers/usb/host/ehci-omap.c
+++ b/drivers/usb/host/ehci-omap.c
@@ -41,6 +41,7 @@
 #include <linux/usb/ulpi.h>
 #include <plat/usb.h>
 #include <linux/regulator/consumer.h>
+#include <linux/pm_runtime.h>
 
 /* EHCI Register Set */
 #define EHCI_INSNREG04                                  (0xA0)
@@ -190,11 +191,8 @@ static int ehci_hcd_omap_probe(struct platform_device *pdev)
                 }
         }
 
-        ret = omap_usbhs_enable(dev);
-        if (ret) {
-                dev_err(dev, "failed to start usbhs with err %d\n", ret);
-                goto err_enable;
-        }
+        pm_runtime_enable(dev);
+        pm_runtime_get_sync(dev);
 
         /*
          * An undocumented "feature" in the OMAP3 EHCI controller,
@@ -240,11 +238,8 @@ static int ehci_hcd_omap_probe(struct platform_device *pdev)
         return 0;
 
 err_add_hcd:
-        omap_usbhs_disable(dev);
-
-err_enable:
         disable_put_regulator(pdata);
-        usb_put_hcd(hcd);
+        pm_runtime_put_sync(dev);
 
 err_io:
         iounmap(regs);
@@ -266,10 +261,12 @@ static int ehci_hcd_omap_remove(struct platform_device *pdev)
         struct usb_hcd *hcd     = dev_get_drvdata(dev);
 
         usb_remove_hcd(hcd);
-        omap_usbhs_disable(dev);
         disable_put_regulator(dev->platform_data);
         iounmap(hcd->regs);
         usb_put_hcd(hcd);
+        pm_runtime_put_sync(dev);
+        pm_runtime_disable(dev);
+
         return 0;
 }
 
diff --git a/drivers/usb/host/ohci-omap3.c b/drivers/usb/host/ohci-omap3.c
index 516ebc4d6cc2..1b8133b6e451 100644
--- a/drivers/usb/host/ohci-omap3.c
+++ b/drivers/usb/host/ohci-omap3.c
@@ -31,6 +31,7 @@
 
 #include <linux/platform_device.h>
 #include <plat/usb.h>
+#include <linux/pm_runtime.h>
 
 /*-------------------------------------------------------------------------*/
 
@@ -134,7 +135,7 @@ static int __devinit ohci_hcd_omap3_probe(struct platform_device *pdev)
         int                     irq;
 
         if (usb_disabled())
-                goto err_end;
+                return -ENODEV;
 
         if (!dev->parent) {
                 dev_err(dev, "Missing parent device\n");
@@ -172,11 +173,8 @@ static int __devinit ohci_hcd_omap3_probe(struct platform_device *pdev)
         hcd->rsrc_len = resource_size(res);
         hcd->regs =  regs;
 
-        ret = omap_usbhs_enable(dev);
-        if (ret) {
-                dev_dbg(dev, "failed to start ohci\n");
-                goto err_end;
-        }
+        pm_runtime_enable(dev);
+        pm_runtime_get_sync(dev);
 
         ohci_hcd_init(hcd_to_ohci(hcd));
 
@@ -189,9 +187,7 @@ static int __devinit ohci_hcd_omap3_probe(struct platform_device *pdev)
         return 0;
 
 err_add_hcd:
-        omap_usbhs_disable(dev);
-
-err_end:
+        pm_runtime_put_sync(dev);
         usb_put_hcd(hcd);
 
 err_io:
@@ -220,9 +216,9 @@ static int __devexit ohci_hcd_omap3_remove(struct platform_device *pdev)
 
         iounmap(hcd->regs);
         usb_remove_hcd(hcd);
-        omap_usbhs_disable(dev);
+        pm_runtime_put_sync(dev);
+        pm_runtime_disable(dev);
         usb_put_hcd(hcd);
-
         return 0;
 }