2 files changed, 3 insertions, 7 deletions

diff --git a/drivers/gpu/nvgpu/tegra/linux/platform_gk20a_tegra.c b/drivers/gpu/nvgpu/tegra/linux/platform_gk20a_tegra.c
index 96312a00..8a83f7d0 100644
--- a/drivers/gpu/nvgpu/tegra/linux/platform_gk20a_tegra.c
+++ b/drivers/gpu/nvgpu/tegra/linux/platform_gk20a_tegra.c
@@ -934,8 +934,6 @@ static int gk20a_tegra_probe(struct device *dev)
         if (tegra_get_chip_id() == TEGRA132)
                 platform->soc_name = "tegra13x";
 
-        platform->g->mm.vidmem_is_vidmem = platform->vidmem_is_vidmem;
-
         gk20a_tegra_get_clocks(dev);
         nvgpu_linux_init_clk_support(platform->g);
         gk20a_tegra_init_secure_alloc(platform->g);
@@ -1051,7 +1049,7 @@ struct gk20a_platform gk20a_tegra_platform = {
 
         .soc_name = "tegra12x",
 
-        .vidmem_is_vidmem = false,
+        .unified_memory = true,
 };
 
 struct gk20a_platform gm20b_tegra_platform = {
@@ -1123,5 +1121,5 @@ struct gk20a_platform gm20b_tegra_platform = {
 
         .soc_name = "tegra21x",
 
-        .vidmem_is_vidmem = false,
+        .unified_memory = true,
 };
diff --git a/drivers/gpu/nvgpu/tegra/linux/platform_gp10b_tegra.c b/drivers/gpu/nvgpu/tegra/linux/platform_gp10b_tegra.c
index b7cbf56e..8b530fac 100644
--- a/drivers/gpu/nvgpu/tegra/linux/platform_gp10b_tegra.c
+++ b/drivers/gpu/nvgpu/tegra/linux/platform_gp10b_tegra.c
@@ -161,8 +161,6 @@ static int gp10b_tegra_probe(struct device *dev)
         platform->g->gr.t18x.ctx_vars.force_preemption_gfxp = false;
         platform->g->gr.t18x.ctx_vars.force_preemption_cilp = false;
 
-        platform->g->mm.vidmem_is_vidmem = platform->vidmem_is_vidmem;
-
         gp10b_tegra_get_clocks(dev);
         nvgpu_linux_init_clk_support(platform->g);
         gk20a_tegra_init_secure_alloc(platform->g);
@@ -436,7 +434,7 @@ struct gk20a_platform gp10b_tegra_platform = {
 
         .soc_name = "tegra18x",
 
-        .vidmem_is_vidmem = false,
+        .unified_memory = true,
 };