3 files changed, 22 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
index 60beab9f..577c8f6e 100644
--- a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
+++ b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
@@ -380,7 +380,7 @@ static const struct gpu_ops vgpu_gv11b_ops = {
                 .pbdma_acquire_val = gk20a_fifo_pbdma_acquire_val,
                 .preempt_channel = vgpu_fifo_preempt_channel,
                 .preempt_tsg = vgpu_fifo_preempt_tsg,
-                .enable_tsg = vgpu_enable_tsg,
+                .enable_tsg = vgpu_gv11b_enable_tsg,
                 .disable_tsg = gk20a_disable_tsg,
                 .tsg_verify_channel_status = NULL,
                 .tsg_verify_status_ctx_reload = NULL,
diff --git a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_tsg_gv11b.c b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_tsg_gv11b.c
index 367c1299..e42bea1f 100644
--- a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_tsg_gv11b.c
+++ b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_tsg_gv11b.c
@@ -58,3 +58,23 @@ int vgpu_gv11b_tsg_bind_channel(struct tsg_gk20a *tsg,
 
         return err;
 }
+
+int vgpu_gv11b_enable_tsg(struct tsg_gk20a *tsg)
+{
+        struct gk20a *g = tsg->g;
+        struct channel_gk20a *ch;
+        struct channel_gk20a *last_ch = NULL;
+
+        nvgpu_rwsem_down_read(&tsg->ch_list_lock);
+        nvgpu_list_for_each_entry(ch, &tsg->ch_list, channel_gk20a, ch_entry) {
+                g->ops.fifo.enable_channel(ch);
+                last_ch = ch;
+        }
+        nvgpu_rwsem_up_read(&tsg->ch_list_lock);
+
+        if (last_ch)
+                g->ops.fifo.ring_channel_doorbell(last_ch);
+
+        return 0;
+}
+
diff --git a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_tsg_gv11b.h b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_tsg_gv11b.h
index 9ce84170..35eb6802 100644
--- a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_tsg_gv11b.h
+++ b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_tsg_gv11b.h
@@ -25,5 +25,5 @@
 
 int vgpu_gv11b_tsg_bind_channel(struct tsg_gk20a *tsg,
                                 struct channel_gk20a *ch);
-
+int vgpu_gv11b_enable_tsg(struct tsg_gk20a *tsg);
 #endif