1 files changed, 3 insertions, 10 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index 7875d621..c0889571 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -1965,20 +1965,13 @@ int gk20a_init_gpu_characteristics(struct gk20a *g)
             gk20a_platform_has_syncpoints(g->dev))
                 gpu->flags |= NVGPU_GPU_FLAGS_HAS_SYNCPOINTS;
 
-        if (IS_ENABLED(CONFIG_GK20A_CYCLE_STATS)) {
-                gpu->flags |= NVGPU_GPU_FLAGS_SUPPORT_CYCLE_STATS;
-                /* the snapshots must be supported only for t210 */
-                if (NVGPU_GPU_ARCH_GM200 == gpu->arch &&
-                                NVGPU_GPU_IMPL_GM20B == gpu->impl) {
-                        gpu->flags |=
-                                NVGPU_GPU_FLAGS_SUPPORT_CYCLE_STATS_SNAPSHOT;
-                }
-        }
-
         gpu->gpc_mask = 1;
 
         g->ops.gr.detect_sm_arch(g);
 
+        if (g->ops.gr.init_cyclestats)
+                g->ops.gr.init_cyclestats(g);
+
         gpu->gpu_ioctl_nr_last = NVGPU_GPU_IOCTL_LAST;
         gpu->tsg_ioctl_nr_last = NVGPU_TSG_IOCTL_LAST;
         gpu->dbg_gpu_ioctl_nr_last = NVGPU_DBG_GPU_IOCTL_LAST;