2 files changed, 3 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gp106/hal_gp106.c b/drivers/gpu/nvgpu/gp106/hal_gp106.c
index e3d5556f..b0eab947 100644
--- a/drivers/gpu/nvgpu/gp106/hal_gp106.c
+++ b/drivers/gpu/nvgpu/gp106/hal_gp106.c
@@ -611,6 +611,7 @@ static const struct gpu_ops gp106_ops = {
                 .pmu_get_queue_tail = pwr_pmu_queue_tail_r,
                 .pmu_lpwr_enable_pg = nvgpu_lpwr_enable_pg,
                 .get_irqdest = gk20a_pmu_get_irqdest,
+                .alloc_super_surface = NULL,
         },
         .clk = {
                 .init_clk_support = gp106_init_clk_support,
diff --git a/drivers/gpu/nvgpu/gp106/sec2_gp106.c b/drivers/gpu/nvgpu/gp106/sec2_gp106.c
index 8e4e5900..08c7f84a 100644
--- a/drivers/gpu/nvgpu/gp106/sec2_gp106.c
+++ b/drivers/gpu/nvgpu/gp106/sec2_gp106.c
@@ -178,6 +178,8 @@ void init_pmu_setup_hw1(struct gk20a *g)
         g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base(pmu);
         g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx(
                 pmu, GK20A_PMU_DMAIDX_VIRT);
+        if (g->ops.pmu_ver.config_pmu_cmdline_args_super_surface)
+                g->ops.pmu_ver.config_pmu_cmdline_args_super_surface(pmu);
 
         nvgpu_flcn_copy_to_dmem(pmu->flcn, g->acr.pmu_args,
                         (u8 *)(g->ops.pmu_ver.get_pmu_cmdline_args_ptr(pmu)),