io_uring/msg_ring: flag target ring as having task_work, if needed

Before the recent change, we didn't even wake the targeted task when
posting the cqe remotely. Now we do wake it, but we do want to ensure
that the recipient knows there's potential work there that needs to
get processed to get the CQE posted.

OR in IORING_SQ_TASKRUN for that purpose.

Link: https://lore.kernel.org/io-uring/[email protected]/
Fixes: 6d043ee ("io_uring: do msg_ring in target task via tw")
Signed-off-by: Jens Axboe <[email protected]>

Author: axboe

io_uring/msg_ring.c

@@ -67,6 +67,7 @@ static int io_msg_ring_data(struct io_kiocb *req)
 				  TWA_SIGNAL_NO_IPI))
 			return -EOWNERDEAD;
 
+		atomic_or(IORING_SQ_TASKRUN, &target_ctx->rings->sq_flags);
 		return IOU_ISSUE_SKIP_COMPLETE;
 	}