1 files changed, 1 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/common/mm/nvgpu_mem.c b/drivers/gpu/nvgpu/common/mm/nvgpu_mem.c
index 2b6e6e6a..9f677058 100644
--- a/drivers/gpu/nvgpu/common/mm/nvgpu_mem.c
+++ b/drivers/gpu/nvgpu/common/mm/nvgpu_mem.c
@@ -62,7 +62,7 @@ void nvgpu_sgt_free(struct nvgpu_sgt *sgt, struct gk20a *g)
 u64 nvgpu_mem_iommu_translate(struct gk20a *g, u64 phys)
 {
         /* ensure it is not vidmem allocation */
-        WARN_ON(is_vidmem_page_alloc(phys));
+        WARN_ON(nvgpu_addr_is_vidmem_page_alloc(phys));
 
         if (nvgpu_iommuable(g) && g->ops.mm.get_iommu_bit)
                 return phys | 1ULL << g->ops.mm.get_iommu_bit(g);