set_irq_wake: fix return code and wake status tracking