1 files changed, 1 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index 2d7266ce..1d594d29 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -397,6 +397,7 @@ int gk20a_init_gpu_characteristics(struct gk20a *g)
 
         __nvgpu_set_enabled(g, NVGPU_SUPPORT_PARTIAL_MAPPINGS, true);
         __nvgpu_set_enabled(g, NVGPU_SUPPORT_MAP_DIRECT_KIND_CTRL, true);
+        __nvgpu_set_enabled(g, NVGPU_SUPPORT_MAP_BUFFER_BATCH, true);
 
         if (IS_ENABLED(CONFIG_SYNC))
                 __nvgpu_set_enabled(g, NVGPU_SUPPORT_SYNC_FENCE_FDS, true);
@@ -464,8 +465,6 @@ int gk20a_init_gpu_characteristics(struct gk20a *g)
         gpu->cbc_cache_line_size = g->gr.cacheline_size;
         gpu->cbc_comptags_per_line = g->gr.comptags_per_cacheline;
 
-        gpu->map_buffer_batch_limit = 256;
-
         if (g->ops.clk.get_maxrate)
                 gpu->max_freq = g->ops.clk.get_maxrate(&g->clk);