1 files changed, 1 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/vgpu.c b/drivers/gpu/nvgpu/vgpu/vgpu.c
index 3c06652b..4948d246 100644
--- a/drivers/gpu/nvgpu/vgpu/vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/vgpu.c
@@ -268,6 +268,7 @@ void vgpu_init_hal_common(struct gk20a *g)
         vgpu_init_mm_ops(gops);
         vgpu_init_debug_ops(gops);
         vgpu_init_fecs_trace_ops(gops);
+        vgpu_init_tsg_ops(gops);
         gops->chip_init_gpu_characteristics = gk20a_init_gpu_characteristics;
 }
 
@@ -340,8 +341,6 @@ int vgpu_pm_finalize_poweron(struct device *dev)
                 goto done;
         }
 
-        g->gpu_characteristics.flags &= ~NVGPU_GPU_FLAGS_SUPPORT_TSG;
-
         gk20a_ctxsw_trace_init(g);
         gk20a_channel_resume(g);