1 files changed, 10 insertions, 5 deletions

diff --git a/drivers/gpu/nvgpu/gv11b/gr_gv11b.c b/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
index 67603739..f8461f9d 100644
--- a/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
@@ -4513,10 +4513,16 @@ int gr_gv11b_decode_priv_addr(struct gk20a *g, u32 addr,
         return 0;
 }
 
-static u32 gr_gv11b_pri_pmmgpc_addr(u32 gpc_num, u32 domain_idx, u32 offset)
+u32 gr_gv11b_get_pmm_per_chiplet_offset(void)
+{
+        return (perf_pmmsys_extent_v() - perf_pmmsys_base_v() + 1);
+}
+
+static u32 gr_gv11b_pri_pmmgpc_addr(struct gk20a *g, u32 gpc_num,
+        u32 domain_idx, u32 offset)
 {
         return perf_pmmgpc_base_v() +
-                (gpc_num * (perf_pmmsys_extent_v() - perf_pmmsys_base_v() + 1)) +
+                (gpc_num * g->ops.gr.get_pmm_per_chiplet_offset()) +
                 (domain_idx * perf_pmmgpc_perdomain_offset_v()) +
                 offset;
 }
@@ -4531,8 +4537,7 @@ static void gr_gv11b_split_pmm_fbp_broadcast_address(struct gk20a *g,
 
         for (fbp_num = 0; fbp_num < g->gr.num_fbps; fbp_num++) {
                 base = perf_pmmfbp_base_v() +
-                        (fbp_num *
-                        (perf_pmmsys_extent_v() - perf_pmmsys_base_v() + 1));
+                        (fbp_num * g->ops.gr.get_pmm_per_chiplet_offset());
 
                 for (domain_idx = domain_start;
                      domain_idx < (domain_start + num_domains);
@@ -4653,7 +4658,7 @@ int gr_gv11b_create_priv_addr_table(struct gk20a *g,
                              domain_idx < (pmm_domain_start + num_domains);
                              domain_idx++) {
                                 priv_addr_table[t++] =
-                                        gr_gv11b_pri_pmmgpc_addr(gpc_num,
+                                        gr_gv11b_pri_pmmgpc_addr(g, gpc_num,
                                         domain_idx, offset);
                         }
                 }