1 files changed, 4 insertions, 4 deletions

diff --git a/drivers/gpu/nvgpu/common/mm/gmmu.c b/drivers/gpu/nvgpu/common/mm/gmmu.c
index 6cca8c2f..8ad7dac7 100644
--- a/drivers/gpu/nvgpu/common/mm/gmmu.c
+++ b/drivers/gpu/nvgpu/common/mm/gmmu.c
@@ -180,7 +180,7 @@ int nvgpu_gmmu_init_page_table(struct vm_gk20a *vm)
                 return err;
 
         /*
-         * One nvgpu_smp_mb() is done after all mapping operations. Don't need
+         * One nvgpu_mb() is done after all mapping operations. Don't need
          * individual barriers for each PD write.
          */
         vm->pdb.mem->skip_wmb = true;
@@ -275,7 +275,7 @@ static int pd_allocate(struct vm_gk20a *vm,
         }
 
         /*
-         * One nvgpu_smp_mb() is done after all mapping operations. Don't need
+         * One nvgpu_mb() is done after all mapping operations. Don't need
          * individual barriers for each PD write.
          */
         pd->mem->skip_wmb = true;
@@ -639,7 +639,7 @@ static int __nvgpu_gmmu_update_page_table(struct vm_gk20a *vm,
                                                 attrs);
 
         unmap_gmmu_pages(g, &vm->pdb);
-        nvgpu_smp_mb();
+        nvgpu_mb();
 
         __gmmu_dbg(g, attrs, "%-5s Done!", sgt ? "MAP" : "UNMAP");
 
@@ -914,7 +914,7 @@ int __nvgpu_set_pte(struct gk20a *g, struct vm_gk20a *vm, u64 vaddr, u32 *pte)
          * There probably also needs to be a TLB invalidate as well but we leave
          * that to the caller of this function.
          */
-        nvgpu_smp_wmb();
+        nvgpu_wmb();
 
         return 0;
 }