1 files changed, 4 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/hal_gp10b.c b/drivers/gpu/nvgpu/gp10b/hal_gp10b.c
index 759d271e..740cb8b7 100644
--- a/drivers/gpu/nvgpu/gp10b/hal_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/hal_gp10b.c
@@ -794,7 +794,8 @@ int gp10b_init_hal(struct gk20a *g)
                         gm20b_flcn_populate_bl_dmem_desc,
                 gops->pmu.update_lspmu_cmdline_args =
                         gm20b_update_lspmu_cmdline_args;
-                gops->pmu.setup_apertures = gm20b_setup_apertures;
+                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;
@@ -806,6 +807,8 @@ int 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;