UPSTREAM: io_uring: allow re-poll if we made progress

commit 10c87333 upstream.

We currently check REQ_F_POLLED before arming async poll for a
notification to retry. If it's set, then we don't allow poll and will
punt to io-wq instead. This is done to prevent a situation where a buggy
driver will repeatedly return that there's space/data available yet we
get -EAGAIN.

However, if we already transferred data, then it should be safe to rely
on poll again. Gate the check on whether or not REQ_F_PARTIAL_IO is
also set.

Change-Id: I5e04e122cdfaa3a754af1fb896568af9ccb5e78b
Signed-off-by: Jens Axboe <axboe@kernel.dk>
Signed-off-by: Sasha Levin <sashal@kernel.org>
(cherry picked from commit a79b13f2)
Bug: 268174392
Signed-off-by: Greg Kroah-Hartman <gregkh@google.com>