1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c b/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c
index 252b4e43..26e4de02 100644
--- a/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c
@@ -821,6 +821,8 @@ void vgpu_init_fifo_ops(struct gpu_ops *gops)
         gops->fifo.setup_ramfc = vgpu_channel_setup_ramfc;
         gops->fifo.preempt_channel = vgpu_fifo_preempt_channel;
         gops->fifo.preempt_tsg = vgpu_fifo_preempt_tsg;
+        gops->fifo.enable_tsg = gk20a_enable_tsg;
+        gops->fifo.disable_tsg = gk20a_disable_tsg;
         gops->fifo.update_runlist = vgpu_fifo_update_runlist;
         gops->fifo.wait_engine_idle = vgpu_fifo_wait_engine_idle;
         gops->fifo.channel_set_priority = vgpu_channel_set_priority;