1 files changed, 6 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 9b3b4ed5..78aef699 100644
--- a/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
+++ b/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
@@ -614,6 +614,10 @@ int vgpu_gp10b_init_hal(struct gk20a *g)
                         gm20b_pmu_populate_loader_cfg,
                 gops->pmu.flcn_populate_bl_dmem_desc =
                         gm20b_flcn_populate_bl_dmem_desc,
+                gops->pmu.update_lspmu_cmdline_args =
+                        gm20b_update_lspmu_cmdline_args;
+                gops->pmu.setup_apertures = gm20b_pmu_setup_apertures;
+                gops->pmu.secured_pmu_start = gm20b_secured_pmu_start;
 
                 gops->pmu.init_wpr_region = gm20b_pmu_init_acr;
                 gops->pmu.load_lsfalcon_ucode = gp10b_load_falcon_ucode;
@@ -625,6 +629,8 @@ int vgpu_gp10b_init_hal(struct gk20a *g)
                 /* Inherit from gk20a */
                 gops->pmu.is_pmu_supported = gk20a_is_pmu_supported,
                 gops->pmu.prepare_ucode = nvgpu_pmu_prepare_ns_ucode_blob,
+                gops->pmu.pmu_setup_hw_and_bootstrap =
+                        gm20b_ns_pmu_setup_hw_and_bootstrap;
                 gops->pmu.pmu_nsbootstrap = pmu_bootstrap,
 
                 gops->pmu.load_lsfalcon_ucode = NULL;