1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/driver_common.c b/drivers/gpu/nvgpu/os/linux/driver_common.c
index c651e394..f1eccd06 100644
--- a/drivers/gpu/nvgpu/os/linux/driver_common.c
+++ b/drivers/gpu/nvgpu/os/linux/driver_common.c
@@ -215,6 +215,8 @@ static void nvgpu_init_mm_vars(struct gk20a *g)
                             platform->unified_memory);
         __nvgpu_set_enabled(g, NVGPU_MM_UNIFY_ADDRESS_SPACES,
                             platform->unify_address_spaces);
+        __nvgpu_set_enabled(g, NVGPU_MM_FORCE_128K_PMU_VM,
+                            platform->force_128K_pmu_vm);
 
         nvgpu_mutex_init(&g->mm.tlb_lock);
         nvgpu_mutex_init(&g->mm.priv_lock);