2 files changed, 3 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 51569fb7..272fadbb 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -341,6 +341,7 @@ struct gpu_ops {
                 void (*write_pm_ptr)(struct gk20a *g,
                                         struct mem_desc *mem, u64 gpu_va);
                 void (*init_elcg_mode)(struct gk20a *g, u32 mode, u32 engine);
+                void (*load_tpc_mask)(struct gk20a *g);
         } gr;
         const char *name;
         struct {
diff --git a/drivers/gpu/nvgpu/gm20b/gr_gm20b.c b/drivers/gpu/nvgpu/gm20b/gr_gm20b.c
index 44a0116b..de3f879a 100644
--- a/drivers/gpu/nvgpu/gm20b/gr_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/gr_gm20b.c
@@ -636,7 +636,7 @@ int gr_gm20b_init_fs_state(struct gk20a *g)
 
         gr_gk20a_init_fs_state(g);
 
-        gr_gm20b_load_tpc_mask(g);
+        g->ops.gr.load_tpc_mask(g);
 
         gk20a_writel(g, gr_bes_zrop_settings_r(),
                      gr_bes_zrop_settings_num_active_ltcs_f(g->ltc_count));
@@ -1605,5 +1605,6 @@ void gm20b_init_gr(struct gpu_ops *gops)
         gops->gr.write_zcull_ptr = gr_gk20a_write_zcull_ptr;
         gops->gr.write_pm_ptr = gr_gk20a_write_pm_ptr;
         gops->gr.init_elcg_mode = gr_gk20a_init_elcg_mode;
+        gops->gr.load_tpc_mask = gr_gm20b_load_tpc_mask;
 
 }