return 0;
 }
 
-static void omap4_keypad_close(struct input_dev *input)
+static void omap4_keypad_stop(struct omap4_keypad *keypad_data)
 {
-       struct omap4_keypad *keypad_data = input_get_drvdata(input);
-
-       disable_irq(keypad_data->irq);
-
        /* Disable interrupts and wake-up events */
        kbd_write_irqreg(keypad_data, OMAP4_KBD_IRQENABLE,
                         OMAP4_VAL_IRQDISABLE);
        /* clear pending interrupts */
        kbd_write_irqreg(keypad_data, OMAP4_KBD_IRQSTATUS,
                         kbd_read_irqreg(keypad_data, OMAP4_KBD_IRQSTATUS));
+}
+
+static void omap4_keypad_close(struct input_dev *input)
+{
+       struct omap4_keypad *keypad_data;
 
+       keypad_data = input_get_drvdata(input);
+       disable_irq(keypad_data->irq);
+       omap4_keypad_stop(keypad_data);
        enable_irq(keypad_data->irq);
 
        pm_runtime_put_sync(input->dev.parent);
        return 0;
 }
 
+static int omap4_keypad_check_revision(struct device *dev,
+                                      struct omap4_keypad *keypad_data)
+{
+       unsigned int rev;
+
+       rev = __raw_readl(keypad_data->base + OMAP4_KBD_REVISION);
+       rev &= 0x03 << 30;
+       rev >>= 30;
+       switch (rev) {
+       case KBD_REVISION_OMAP4:
+               keypad_data->reg_offset = 0x00;
+               keypad_data->irqreg_offset = 0x00;
+               break;
+       case KBD_REVISION_OMAP5:
+               keypad_data->reg_offset = 0x10;
+               keypad_data->irqreg_offset = 0x0c;
+               break;
+       default:
+               dev_err(dev, "Keypad reports unsupported revision %d", rev);
+               return -EINVAL;
+       }
+
+       return 0;
+}
+
 static int omap4_keypad_probe(struct platform_device *pdev)
 {
        struct omap4_keypad *keypad_data;
        struct input_dev *input_dev;
        struct resource *res;
        unsigned int max_keys;
-       int rev;
        int irq;
        int error;
 
                goto err_release_mem;
        }
 
+       pm_runtime_enable(&pdev->dev);
 
        /*
         * Enable clocks for the keypad module so that we can read
         * revision register.
         */
-       pm_runtime_enable(&pdev->dev);
        error = pm_runtime_get_sync(&pdev->dev);
        if (error) {
                dev_err(&pdev->dev, "pm_runtime_get_sync() failed\n");
-               goto err_unmap;
-       }
-       rev = __raw_readl(keypad_data->base + OMAP4_KBD_REVISION);
-       rev &= 0x03 << 30;
-       rev >>= 30;
-       switch (rev) {
-       case KBD_REVISION_OMAP4:
-               keypad_data->reg_offset = 0x00;
-               keypad_data->irqreg_offset = 0x00;
-               break;
-       case KBD_REVISION_OMAP5:
-               keypad_data->reg_offset = 0x10;
-               keypad_data->irqreg_offset = 0x0c;
-               break;
-       default:
-               dev_err(&pdev->dev,
-                       "Keypad reports unsupported revision %d", rev);
-               error = -EINVAL;
-               goto err_pm_put_sync;
+               pm_runtime_put_noidle(&pdev->dev);
+       } else {
+               error = omap4_keypad_check_revision(&pdev->dev,
+                                                   keypad_data);
+               if (!error) {
+                       /* Ensure device does not raise interrupts */
+                       omap4_keypad_stop(keypad_data);
+               }
+               pm_runtime_put_sync(&pdev->dev);
        }
+       if (error)
+               goto err_pm_disable;
 
        /* input device allocation */
        keypad_data->input = input_dev = input_allocate_device();
        if (!input_dev) {
                error = -ENOMEM;
-               goto err_pm_put_sync;
+               goto err_pm_disable;
        }
 
        input_dev->name = pdev->name;
                goto err_free_keymap;
        }
 
-       device_init_wakeup(&pdev->dev, true);
-       pm_runtime_put_sync(&pdev->dev);
-
        error = input_register_device(keypad_data->input);
        if (error < 0) {
                dev_err(&pdev->dev, "failed to register input device\n");
-               goto err_pm_disable;
+               goto err_free_irq;
        }
 
+       device_init_wakeup(&pdev->dev, true);
        platform_set_drvdata(pdev, keypad_data);
+
        return 0;
 
-err_pm_disable:
-       pm_runtime_disable(&pdev->dev);
+err_free_irq:
        free_irq(keypad_data->irq, keypad_data);
 err_free_keymap:
        kfree(keypad_data->keymap);
 err_free_input:
        input_free_device(input_dev);
-err_pm_put_sync:
-       pm_runtime_put_sync(&pdev->dev);
-err_unmap:
+err_pm_disable:
+       pm_runtime_disable(&pdev->dev);
        iounmap(keypad_data->base);
 err_release_mem:
        release_mem_region(res->start, resource_size(res));