1 files changed, 3 insertions, 2 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;
 }