1 files changed, 3 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/cde.c b/drivers/gpu/nvgpu/os/linux/cde.c
index a15ef560..7b2cba7d 100644
--- a/drivers/gpu/nvgpu/os/linux/cde.c
+++ b/drivers/gpu/nvgpu/os/linux/cde.c
@@ -1272,7 +1272,7 @@ __releases(&cde_app->mutex)
                 nvgpu_log_info(g, "double finish cde context %p on channel %p",
                                 cde_ctx, ch);
 
-        if (ch->has_timedout) {
+        if (gk20a_channel_check_timedout(ch)) {
                 if (cde_ctx->is_temporary) {
                         nvgpu_warn(g,
                                         "cde: channel had timed out"
@@ -1299,8 +1299,9 @@ __releases(&cde_app->mutex)
                         msecs_to_jiffies(CTX_DELETE_TIME));
         }
 
-        if (!ch->has_timedout)
+        if (!gk20a_channel_check_timedout(ch)) {
                 gk20a_cde_ctx_release(cde_ctx);
+        }
 }
 
 static int gk20a_cde_load(struct gk20a_cde_ctx *cde_ctx)