1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/vgpu.c b/drivers/gpu/nvgpu/vgpu/vgpu.c
index b67f4d95..90197ab4 100644
--- a/drivers/gpu/nvgpu/vgpu/vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/vgpu.c
@@ -22,6 +22,7 @@
 #include "gk20a/debug_gk20a.h"
 #include "gk20a/hal_gk20a.h"
 #include "gk20a/hw_mc_gk20a.h"
+#include "gk20a/ctxsw_trace_gk20a.h"
 #include "gm20b/hal_gm20b.h"
 
 #ifdef CONFIG_ARCH_TEGRA_18x_SOC
@@ -120,6 +121,12 @@ static int vgpu_intr_thread(void *dev_id)
                         break;
                 }
 
+                if (msg->event == TEGRA_VGPU_EVENT_FECS_TRACE) {
+                        vgpu_fecs_trace_data_update(g);
+                        tegra_gr_comm_release(handle);
+                        continue;
+                }
+
                 if (msg->unit == TEGRA_VGPU_INTR_GR)
                         vgpu_gr_isr(g, &msg->info.gr_intr);
                 else if (msg->unit == TEGRA_VGPU_NONSTALL_INTR_GR)
@@ -334,6 +341,7 @@ int vgpu_pm_finalize_poweron(struct device *dev)
 
         g->gpu_characteristics.flags &= ~NVGPU_GPU_FLAGS_SUPPORT_TSG;
 
+        gk20a_ctxsw_trace_init(g);
         gk20a_channel_resume(g);
 
 done: