1 files changed, 3 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/vgpu.c b/drivers/gpu/nvgpu/vgpu/vgpu.c
index a88d9e09..02cc5b47 100644
--- a/drivers/gpu/nvgpu/vgpu/vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/vgpu.c
@@ -245,6 +245,9 @@ static void vgpu_init_vars(struct gk20a *g, struct gk20a_platform *platform)
         g->ptimer_src_freq = platform->ptimer_src_freq;
         g->can_railgate = platform->can_railgate_init;
         g->railgate_delay = platform->railgate_delay_init;
+
+        __nvgpu_set_enabled(g, NVGPU_MM_UNIFY_ADDRESS_SPACES,
+                            platform->unify_address_spaces);
 }
 
 static int vgpu_init_support(struct platform_device *pdev)