gpu: nvgpu: gm20b: add gr ops for load tpc mask

gr_fe_tpc_fs_r addr is different for t19x

Change-Id: I9868fcaf94d063753c4ecf1970b53374cb67b380
Signed-off-by: Seema Khowala <seemaj@nvidia.com>
Reviewed-on: http://git-master/r/1310326
Reviewed-by: Automatic_Commit_Validation_User
Reviewed-by: svccoveritychecker <svccoveritychecker@nvidia.com>
GVS: Gerrit_Virtual_Submit
Reviewed-by: Seshendra Gadagottu <sgadagottu@nvidia.com>
Reviewed-by: Terje Bergstrom <tbergstrom@nvidia.com>

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;
 
 }