#include <linux/of.h>
 #include <linux/of_device.h>
 #include <linux/sort.h>
+#include <linux/pm_wakeirq.h>
 
 #include <linux/mfd/ti_am335x_tscadc.h>
 
        }
 
        device_init_wakeup(&pdev->dev, true);
+       err = dev_pm_set_wake_irq(&pdev->dev, ts_dev->irq);
+       if (err)
+               dev_err(&pdev->dev, "irq wake enable failed.\n");
 
        titsc_writel(ts_dev, REG_IRQSTATUS, TSC_IRQENB_MASK);
        titsc_writel(ts_dev, REG_IRQENABLE, IRQENB_FIFO0THRES);
        return 0;
 
 err_free_irq:
+       dev_pm_clear_wake_irq(&pdev->dev);
        device_init_wakeup(&pdev->dev, false);
        free_irq(ts_dev->irq, ts_dev);
 err_free_mem:
        struct titsc *ts_dev = platform_get_drvdata(pdev);
        u32 steps;
 
+       dev_pm_clear_wake_irq(&pdev->dev);
        device_init_wakeup(&pdev->dev, false);
        free_irq(ts_dev->irq, ts_dev);