2 files changed, 6 insertions, 4 deletions

diff --git a/drivers/gpu/nvgpu/common/mm/gmmu.c b/drivers/gpu/nvgpu/common/mm/gmmu.c
index 4e723e2e..f6f2b9ad 100644
--- a/drivers/gpu/nvgpu/common/mm/gmmu.c
+++ b/drivers/gpu/nvgpu/common/mm/gmmu.c
@@ -215,10 +215,11 @@ static u64 nvgpu_pde_phys_addr(struct gk20a *g, struct nvgpu_gmmu_pd *pd)
 {
         u64 page_addr;
 
-        if (g->mm.has_physical_mode)
+        if (nvgpu_is_enabled(g, NVGPU_SUPPORT_NVLINK)) {
                 page_addr = nvgpu_mem_get_phys_addr(g, pd->mem);
-        else
+        } else {
                 page_addr = nvgpu_mem_get_addr(g, pd->mem);
+        }
 
         return page_addr + pd->mem_offs;
 }
diff --git a/drivers/gpu/nvgpu/common/mm/mm.c b/drivers/gpu/nvgpu/common/mm/mm.c
index 42d708ee..2c3a1cd6 100644
--- a/drivers/gpu/nvgpu/common/mm/mm.c
+++ b/drivers/gpu/nvgpu/common/mm/mm.c
@@ -128,10 +128,11 @@ int nvgpu_mm_suspend(struct gk20a *g)
 
 u64 nvgpu_inst_block_addr(struct gk20a *g, struct nvgpu_mem *inst_block)
 {
-        if (g->mm.has_physical_mode)
+        if (nvgpu_is_enabled(g, NVGPU_SUPPORT_NVLINK)) {
                 return nvgpu_mem_get_phys_addr(g, inst_block);
-        else
+        } else {
                 return nvgpu_mem_get_addr(g, inst_block);
+        }
 }
 
 void nvgpu_free_inst_block(struct gk20a *g, struct nvgpu_mem *inst_block)