1 files changed, 5 insertions, 5 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index 3875ec5c..d4d6cd2d 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -1738,7 +1738,7 @@ int gr_gk20a_update_hwpm_ctxsw_mode(struct gk20a *g,
                         pm_ctx->mem.gpu_va = nvgpu_gmmu_map(c->vm,
                                                         &pm_ctx->mem,
                                                         pm_ctx->mem.size,
-                                                        NVGPU_MAP_BUFFER_FLAGS_CACHEABLE_TRUE,
+                                                        NVGPU_AS_MAP_BUFFER_FLAGS_CACHEABLE,
                                                         gk20a_mem_flag_none, true,
                                                         pm_ctx->mem.aperture);
                         if (!pm_ctx->mem.gpu_va) {
@@ -2633,7 +2633,7 @@ static int gr_gk20a_map_global_ctx_buffers(struct gk20a *g,
         }
 
         gpu_va = nvgpu_gmmu_map(ch_vm, mem, mem->size,
-                                NVGPU_MAP_BUFFER_FLAGS_CACHEABLE_TRUE,
+                                NVGPU_AS_MAP_BUFFER_FLAGS_CACHEABLE,
                                 gk20a_mem_flag_none, true, mem->aperture);
         if (!gpu_va)
                 goto clean_up;
@@ -2651,7 +2651,7 @@ static int gr_gk20a_map_global_ctx_buffers(struct gk20a *g,
         }
 
         gpu_va = nvgpu_gmmu_map(ch_vm, mem, mem->size,
-                                NVGPU_MAP_BUFFER_FLAGS_CACHEABLE_TRUE,
+                                NVGPU_AS_MAP_BUFFER_FLAGS_CACHEABLE,
                                 gk20a_mem_flag_none, false, mem->aperture);
         if (!gpu_va)
                 goto clean_up;
@@ -2669,7 +2669,7 @@ static int gr_gk20a_map_global_ctx_buffers(struct gk20a *g,
         }
 
         gpu_va = nvgpu_gmmu_map(ch_vm, mem, mem->size,
-                                NVGPU_MAP_BUFFER_FLAGS_CACHEABLE_TRUE,
+                                NVGPU_AS_MAP_BUFFER_FLAGS_CACHEABLE,
                                 gk20a_mem_flag_none, true, mem->aperture);
         if (!gpu_va)
                 goto clean_up;
@@ -2736,7 +2736,7 @@ int gr_gk20a_alloc_gr_ctx(struct gk20a *g,
         gr_ctx->mem.gpu_va = nvgpu_gmmu_map(vm,
                                         &gr_ctx->mem,
                                         gr_ctx->mem.size,
-                                        NVGPU_MAP_BUFFER_FLAGS_CACHEABLE_FALSE,
+                                        0, /* not GPU-cacheable */
                                         gk20a_mem_flag_none, true,
                                         gr_ctx->mem.aperture);
         if (!gr_ctx->mem.gpu_va)