2 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gm20b/fifo_gm20b.c b/drivers/gpu/nvgpu/gm20b/fifo_gm20b.c
index eaa22dc2..bb873bcc 100644
--- a/drivers/gpu/nvgpu/gm20b/fifo_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/fifo_gm20b.c
@@ -142,6 +142,7 @@ void gm20b_init_fifo(struct gpu_ops *gops)
         gops->fifo.channel_set_timeslice = gk20a_channel_set_timeslice;
 
         gops->fifo.preempt_channel = gk20a_fifo_preempt_channel;
+        gops->fifo.preempt_tsg = gk20a_fifo_preempt_tsg;
         gops->fifo.update_runlist = gk20a_fifo_update_runlist;
         gops->fifo.trigger_mmu_fault = gm20b_fifo_trigger_mmu_fault;
         gops->fifo.wait_engine_idle = gk20a_fifo_wait_engine_idle;
diff --git a/drivers/gpu/nvgpu/gm20b/hal_gm20b.c b/drivers/gpu/nvgpu/gm20b/hal_gm20b.c
index df25be5e..c0113498 100644
--- a/drivers/gpu/nvgpu/gm20b/hal_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/hal_gm20b.c
@@ -211,6 +211,7 @@ int gm20b_init_hal(struct gk20a *g)
         gm20b_init_debug_ops(gops);
         gm20b_init_cde_ops(gops);
         gm20b_init_therm_ops(gops);
+        gk20a_init_tsg_ops(gops);
         gops->name = "gm20b";
         gops->chip_init_gpu_characteristics = gk20a_init_gpu_characteristics;
         gops->get_litter_value = gm20b_get_litter_value;