2 files changed, 3 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gm20b/acr_gm20b.c b/drivers/gpu/nvgpu/gm20b/acr_gm20b.c
index 470a93bc..cb874a48 100644
--- a/drivers/gpu/nvgpu/gm20b/acr_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/acr_gm20b.c
@@ -258,7 +258,8 @@ int prepare_ucode_blob(struct gk20a *g)
                 gm20b_dbg_pmu("managed LS falcon %d, WPR size %d bytes.\n",
                         plsfm->managed_flcn_cnt, plsfm->wpr_size);
                 lsfm_init_wpr_contents(g, plsfm, nonwpr_addr);
-                g->acr.ucode_blob_start = NV_MC_SMMU_VADDR_TRANSLATE(iova);
+                g->acr.ucode_blob_start =
+                        gk20a_mm_smmu_vaddr_translate(g, iova);
                 g->acr.ucode_blob_size = plsfm->wpr_size;
                 gm20b_dbg_pmu("base reg carveout 2:%x\n",
                 readl(mc + MC_SECURITY_CARVEOUT2_BOM_0));
diff --git a/drivers/gpu/nvgpu/gm20b/mm_gm20b.c b/drivers/gpu/nvgpu/gm20b/mm_gm20b.c
index 678ef4fd..0113e227 100644
--- a/drivers/gpu/nvgpu/gm20b/mm_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/mm_gm20b.c
@@ -301,4 +301,5 @@ void gm20b_init_mm(struct gpu_ops *gops)
         gops->mm.tlb_invalidate = gk20a_mm_tlb_invalidate;
         gops->mm.set_big_page_size = gm20b_mm_set_big_page_size;
         gops->mm.get_big_page_sizes = gm20b_mm_get_big_page_sizes;
+        gops->mm.get_physical_addr_bits = gk20a_mm_get_physical_addr_bits;
 }