1 files changed, 4 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gv11b/hal_gv11b.c b/drivers/gpu/nvgpu/gv11b/hal_gv11b.c
index 168def5c..22beabf7 100644
--- a/drivers/gpu/nvgpu/gv11b/hal_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/hal_gv11b.c
@@ -68,6 +68,7 @@
 
 #include "dbg_gpu_gv11b.h"
 #include "hal_gv11b.h"
+#include "css_gr_gv11b.h"
 #include "gr_gv11b.h"
 #include "mc_gv11b.h"
 #include "ltc_gv11b.h"
@@ -647,9 +648,9 @@ static const struct gpu_ops gv11b_ops = {
         },
 #if defined(CONFIG_GK20A_CYCLE_STATS)
         .css = {
-                .enable_snapshot = css_hw_enable_snapshot,
-                .disable_snapshot = css_hw_disable_snapshot,
-                .check_data_available = css_hw_check_data_available,
+                .enable_snapshot = gv11b_css_hw_enable_snapshot,
+                .disable_snapshot = gv11b_css_hw_disable_snapshot,
+                .check_data_available = gv11b_css_hw_check_data_available,
                 .set_handled_snapshots = css_hw_set_handled_snapshots,
                 .allocate_perfmon_ids = css_gr_allocate_perfmon_ids,
                 .release_perfmon_ids = css_gr_release_perfmon_ids,