#include <linux/input/touchscreen.h>
 #include <linux/acpi.h>
 #include <linux/of.h>
+#include <linux/pm_wakeirq.h>
 #include <linux/gpio/consumer.h>
 #include <linux/regulator/consumer.h>
 #include <linux/uuid.h>
        u8 cmd_resp[HEADER_SIZE];
        struct completion cmd_done;
 
-       bool wake_irq_enabled;
        bool keep_power_in_suspend;
 
        /* Must be last to be used for DMA operations */
        if (!client->dev.of_node)
                device_init_wakeup(&client->dev, true);
 
+       /*
+        * The wake IRQ should be declared via device tree instead of assuming
+        * the IRQ can wake the system. This is here for legacy reasons and
+        * will be removed once the i2c-core supports querying ACPI for wake
+        * capabilities.
+        */
+       if (!client->dev.power.wakeirq)
+               dev_pm_set_wake_irq(&client->dev, client->irq);
+
        error = devm_device_add_group(&client->dev, &elants_attribute_group);
        if (error) {
                dev_err(&client->dev, "failed to create sysfs attributes: %d\n",
                 * The device will automatically enter idle mode
                 * that has reduced power consumption.
                 */
-               ts->wake_irq_enabled = (enable_irq_wake(client->irq) == 0);
+               return 0;
        } else if (ts->keep_power_in_suspend) {
                for (retry_cnt = 0; retry_cnt < MAX_RETRIES; retry_cnt++) {
                        error = elants_i2c_send(client, set_sleep_cmd,
        int error;
 
        if (device_may_wakeup(dev)) {
-               if (ts->wake_irq_enabled)
-                       disable_irq_wake(client->irq);
                elants_i2c_sw_reset(client);
        } else if (ts->keep_power_in_suspend) {
                for (retry_cnt = 0; retry_cnt < MAX_RETRIES; retry_cnt++) {