1 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/platform_gp10b_tegra.c b/drivers/gpu/nvgpu/gp10b/platform_gp10b_tegra.c
index 587e0a7a..64d9d917 100644
--- a/drivers/gpu/nvgpu/gp10b/platform_gp10b_tegra.c
+++ b/drivers/gpu/nvgpu/gp10b/platform_gp10b_tegra.c
@@ -168,6 +168,8 @@ static int gp10b_tegra_probe(struct device *dev)
                         &platform->g->gr.t18x.
                                 ctx_vars.dump_ctxsw_stats_on_channel_close);
 
+        platform->g->mm.vidmem_is_vidmem = platform->vidmem_is_vidmem;
+
         gp10b_tegra_get_clocks(dev);
 
         return 0;
@@ -417,6 +419,8 @@ struct gk20a_platform t18x_gpu_tegra_platform = {
         .force_reset_in_do_idle = false,
 
         .soc_name = "tegra18x",
+
+        .vidmem_is_vidmem = false,
 };