diff --git a/drivers/mailbox/rockchip-mailbox.c b/drivers/mailbox/rockchip-mailbox.c
index 99b9421b8494..1e0f867cd6e1 100644
--- a/drivers/mailbox/rockchip-mailbox.c
+++ b/drivers/mailbox/rockchip-mailbox.c
@@ -224,6 +224,9 @@ static int rockchip_mbox_probe(struct platform_device *pdev)
 							mb);
 			if (ret < 0)
 				return ret;
+
+			if (device_property_present(&pdev->dev, "wakeup-source"))
+				enable_irq_wake(irq);
 		}
 
 		mb->chans[i].idx = i;