4 files changed, 11 insertions, 9 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/channel_gk20a.c b/drivers/gpu/nvgpu/gk20a/channel_gk20a.c
index 7ec5ade4..b480c80a 100644
--- a/drivers/gpu/nvgpu/gk20a/channel_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/channel_gk20a.c
@@ -392,7 +392,7 @@ void channel_gk20a_disable(struct channel_gk20a *ch)
                         ccsr_channel_enable_clr_true_f());
 }
 
-void gk20a_channel_abort(struct channel_gk20a *ch)
+void gk20a_channel_abort(struct channel_gk20a *ch, bool channel_preempt)
 {
         struct channel_gk20a_job *job, *n;
         bool released_job_semaphore = false;
@@ -404,6 +404,9 @@ void gk20a_channel_abort(struct channel_gk20a *ch)
 
         ch->g->ops.fifo.disable_channel(ch);
 
+        if (channel_preempt)
+                ch->g->ops.fifo.preempt_channel(ch->g, ch->hw_chid);
+
         /* ensure no fences are pending */
         mutex_lock(&ch->sync_lock);
         if (ch->sync)
@@ -455,8 +458,7 @@ int gk20a_wait_channel_idle(struct channel_gk20a *ch)
 
 void gk20a_disable_channel(struct channel_gk20a *ch)
 {
-        gk20a_channel_abort(ch);
-        ch->g->ops.fifo.preempt_channel(ch->g, ch->hw_chid);
+        gk20a_channel_abort(ch, true);
         channel_gk20a_update_runlist(ch, false);
 }
 
@@ -1621,7 +1623,7 @@ static void gk20a_channel_timeout_handler(struct work_struct *work)
                         gk20a_fifo_abort_tsg(g, ch->tsgid);
                 } else {
                         gk20a_fifo_set_ctx_mmu_error_ch(g, ch);
-                        gk20a_channel_abort(ch);
+                        gk20a_channel_abort(ch, false);
                 }
         } else {
                 /* If failing engine, trigger recovery */
diff --git a/drivers/gpu/nvgpu/gk20a/channel_gk20a.h b/drivers/gpu/nvgpu/gk20a/channel_gk20a.h
index cb8ffa95..f62d2731 100644
--- a/drivers/gpu/nvgpu/gk20a/channel_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/channel_gk20a.h
@@ -195,7 +195,7 @@ void gk20a_channel_close(struct channel_gk20a *ch);
 bool gk20a_channel_update_and_check_timeout(struct channel_gk20a *ch,
                                             u32 timeout_delta_ms);
 void gk20a_disable_channel(struct channel_gk20a *ch);
-void gk20a_channel_abort(struct channel_gk20a *ch);
+void gk20a_channel_abort(struct channel_gk20a *ch, bool channel_preempt);
 int gk20a_channel_finish(struct channel_gk20a *ch, unsigned long timeout);
 void gk20a_set_error_notifier(struct channel_gk20a *ch, __u32 error);
 void gk20a_channel_semaphore_wakeup(struct gk20a *g);
diff --git a/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c b/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
index 11fcc805..1727cf1d 100644
--- a/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
@@ -894,7 +894,7 @@ void gk20a_fifo_abort_tsg(struct gk20a *g, u32 tsgid)
         mutex_lock(&tsg->ch_list_lock);
         list_for_each_entry(ch, &tsg->ch_list, ch_entry) {
                 if (gk20a_channel_get(ch)) {
-                        gk20a_channel_abort(ch);
+                        gk20a_channel_abort(ch, false);
                         gk20a_channel_put(ch);
                 }
         }
@@ -1064,7 +1064,7 @@ static bool gk20a_fifo_handle_mmu_fault(
                         if (referenced_channel) {
                                 if (!g->fifo.deferred_reset_pending)
                                         verbose = gk20a_fifo_set_ctx_mmu_error_ch(g, ch);
-                                gk20a_channel_abort(ch);
+                                gk20a_channel_abort(ch, false);
                                 gk20a_channel_put(ch);
                         } else {
                                 gk20a_err(dev_from_gk20a(g),
@@ -1217,7 +1217,7 @@ void gk20a_fifo_recover_ch(struct gk20a *g, u32 hw_chid, bool verbose)
                 struct channel_gk20a *ch = &g->fifo.channel[hw_chid];
 
                 if (gk20a_channel_get(ch)) {
-                        gk20a_channel_abort(ch);
+                        gk20a_channel_abort(ch, false);
 
                         if (gk20a_fifo_set_ctx_mmu_error_ch(g, ch))
                                 gk20a_debug_dump(g->dev);
diff --git a/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c b/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c
index dd6630a7..3db215bc 100644
--- a/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c
@@ -566,7 +566,7 @@ int vgpu_fifo_isr(struct gk20a *g, struct tegra_vgpu_fifo_intr_info *info)
                 break;
         case TEGRA_VGPU_FIFO_INTR_MMU_FAULT:
                 vgpu_fifo_set_ctx_mmu_error(g, ch);
-                gk20a_channel_abort(ch);
+                gk20a_channel_abort(ch, false);
                 break;
         default:
                 WARN_ON(1);