1 files changed, 11 insertions, 6 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/ioctl_channel.c b/drivers/gpu/nvgpu/os/linux/ioctl_channel.c
index 45d49474..d0d4b1af 100644
--- a/drivers/gpu/nvgpu/os/linux/ioctl_channel.c
+++ b/drivers/gpu/nvgpu/os/linux/ioctl_channel.c
@@ -636,8 +636,9 @@ static int gk20a_channel_wait_semaphore(struct channel_gk20a *ch,
         int ret = 0;
 
         /* do not wait if channel has timed out */
-        if (ch->has_timedout)
+        if (gk20a_channel_check_timedout(ch)) {
                 return -ETIMEDOUT;
+        }
 
         dmabuf = dma_buf_get(id);
         if (IS_ERR(dmabuf)) {
@@ -656,7 +657,8 @@ static int gk20a_channel_wait_semaphore(struct channel_gk20a *ch,
 
         ret = NVGPU_COND_WAIT_INTERRUPTIBLE(
                         &ch->semaphore_wq,
-                        *semaphore == payload || ch->has_timedout,
+                        *semaphore == payload ||
+                        gk20a_channel_check_timedout(ch),
                         timeout);
 
         dma_buf_kunmap(dmabuf, offset >> PAGE_SHIFT, data);
@@ -680,8 +682,9 @@ static int gk20a_channel_wait(struct channel_gk20a *ch,
 
         nvgpu_log_fn(g, " ");
 
-        if (ch->has_timedout)
+        if (gk20a_channel_check_timedout(ch)) {
                 return -ETIMEDOUT;
+        }
 
         switch (args->type) {
         case NVGPU_WAIT_TYPE_NOTIFIER:
@@ -716,7 +719,8 @@ static int gk20a_channel_wait(struct channel_gk20a *ch,
                  * calling this ioctl */
                 remain = NVGPU_COND_WAIT_INTERRUPTIBLE(
                                 &ch->notifier_wq,
-                                notif->status == 0 || ch->has_timedout,
+                                notif->status == 0 ||
+                                gk20a_channel_check_timedout(ch),
                                 args->timeout);
 
                 if (remain == 0 && notif->status != 0) {
@@ -786,8 +790,9 @@ static int gk20a_ioctl_channel_submit_gpfifo(
         profile = gk20a_fifo_profile_acquire(ch->g);
         gk20a_fifo_profile_snapshot(profile, PROFILE_IOCTL_ENTRY);
 
-        if (ch->has_timedout)
+        if (gk20a_channel_check_timedout(ch)) {
                 return -ETIMEDOUT;
+        }
 
         nvgpu_get_fence_args(&args->fence, &fence);
         submit_flags =
@@ -1249,7 +1254,7 @@ long gk20a_channel_ioctl(struct file *filp,
         }
         case NVGPU_IOCTL_CHANNEL_GET_TIMEDOUT:
                 ((struct nvgpu_get_param_args *)buf)->value =
-                        ch->has_timedout;
+                        gk20a_channel_check_timedout(ch);
                 break;
         case NVGPU_IOCTL_CHANNEL_ENABLE:
                 err = gk20a_busy(ch->g);