2 files changed, 3 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/ctxsw_trace_gk20a.c b/drivers/gpu/nvgpu/gk20a/ctxsw_trace_gk20a.c
index ffd15a37..4e265b81 100644
--- a/drivers/gpu/nvgpu/gk20a/ctxsw_trace_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/ctxsw_trace_gk20a.c
@@ -518,9 +518,6 @@ int gk20a_ctxsw_trace_init(struct gk20a *g)
         if (unlikely(err))
                 goto fail;
 
-        g->gpu_characteristics.flags |=
-                NVGPU_GPU_FLAGS_SUPPORT_FECS_CTXSW_TRACE;
-
         return 0;
 
 fail:
diff --git a/drivers/gpu/nvgpu/gk20a/fecs_trace_gk20a.c b/drivers/gpu/nvgpu/gk20a/fecs_trace_gk20a.c
index 8244403e..05c53a66 100644
--- a/drivers/gpu/nvgpu/gk20a/fecs_trace_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/fecs_trace_gk20a.c
@@ -584,6 +584,9 @@ static int gk20a_fecs_trace_init(struct gk20a *g)
         nvgpu_mutex_init(&trace->hash_lock);
         hash_init(trace->pid_hash_table);
 
+        g->gpu_characteristics.flags |=
+                NVGPU_GPU_FLAGS_SUPPORT_FECS_CTXSW_TRACE;
+
         gk20a_fecs_trace_debugfs_init(g);
         return 0;