1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/common/mm/gmmu.c b/drivers/gpu/nvgpu/common/mm/gmmu.c
index 22f73083..1455822c 100644
--- a/drivers/gpu/nvgpu/common/mm/gmmu.c
+++ b/drivers/gpu/nvgpu/common/mm/gmmu.c
@@ -533,7 +533,8 @@ static int __nvgpu_gmmu_do_update_page_table(struct vm_gk20a *vm,
                         continue;
                 }
 
-                phys_addr = nvgpu_sgt_get_phys(sgt, sgl) + space_to_skip;
+                phys_addr = g->ops.mm.gpu_phys_addr(g, attrs,
+                            nvgpu_sgt_get_phys(sgt, sgl)) + space_to_skip;
                 chunk_length = min(length,
                         nvgpu_sgt_get_length(sgt, sgl) - space_to_skip);