2 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c b/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
index c56e900e..53ba1e14 100644
--- a/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
+++ b/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
@@ -227,6 +227,8 @@ static const struct gpu_ops vgpu_gp10b_ops = {
                 .add_ctxsw_reg_perf_pma = gr_gk20a_add_ctxsw_reg_perf_pma,
                 .decode_priv_addr = gr_gk20a_decode_priv_addr,
                 .create_priv_addr_table = gr_gk20a_create_priv_addr_table,
+                .get_pmm_per_chiplet_offset =
+                        gr_gm20b_get_pmm_per_chiplet_offset,
         },
         .fb = {
                 .reset = fb_gk20a_reset,
diff --git a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
index a4496044..0b9049dd 100644
--- a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
+++ b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
@@ -263,6 +263,8 @@ static const struct gpu_ops vgpu_gv11b_ops = {
                 .add_ctxsw_reg_perf_pma = gr_gk20a_add_ctxsw_reg_perf_pma,
                 .decode_priv_addr = gr_gv11b_decode_priv_addr,
                 .create_priv_addr_table = gr_gv11b_create_priv_addr_table,
+                .get_pmm_per_chiplet_offset =
+                        gr_gv11b_get_pmm_per_chiplet_offset,
         },
         .fb = {
                 .reset = gv11b_fb_reset,