1 files changed, 3 insertions, 3 deletions

diff --git a/fs/aio.c b/fs/aio.c
index ea2e19820381..d02f43b50a3d 100644
--- a/fs/aio.c
+++ b/fs/aio.c
@@ -303,7 +303,7 @@ static void wait_for_all_aios(struct kioctx *ctx)
         set_task_state(tsk, TASK_UNINTERRUPTIBLE);
         while (ctx->reqs_active) {
                 spin_unlock_irq(&ctx->ctx_lock);
-                schedule();
+                io_schedule();
                 set_task_state(tsk, TASK_UNINTERRUPTIBLE);
                 spin_lock_irq(&ctx->ctx_lock);
         }
@@ -323,7 +323,7 @@ ssize_t fastcall wait_on_sync_kiocb(struct kiocb *iocb)
                 set_current_state(TASK_UNINTERRUPTIBLE);
                 if (!iocb->ki_users)
                         break;
-                schedule();
+                io_schedule();
         }
         __set_current_state(TASK_RUNNING);
         return iocb->ki_user_data;
@@ -1170,7 +1170,7 @@ retry:
                         ret = 0;
                         if (to.timed_out)       /* Only check after read evt */
                                 break;
-                        schedule();
+                        io_schedule();
                         if (signal_pending(tsk)) {
                                 ret = -EINTR;
                                 break;