diff --git a/drivers/gpu/nvgpu/os/linux/platform_gp10b_tegra.c b/drivers/gpu/nvgpu/os/linux/platform_gp10b_tegra.c
index a8308496a..41f96935a 100644
--- a/drivers/gpu/nvgpu/os/linux/platform_gp10b_tegra.c
+++ b/drivers/gpu/nvgpu/os/linux/platform_gp10b_tegra.c
@@ -491,6 +491,7 @@ struct gk20a_platform gp10b_tegra_platform = {
 	.soc_name = "tegra18x",
 
 	.unified_memory = true,
+	.unify_address_spaces = true,
 	.dma_mask = DMA_BIT_MASK(36),
 
 	.ltc_streamid = TEGRA_SID_GPUB,
diff --git a/drivers/gpu/nvgpu/os/linux/platform_gv11b_tegra.c b/drivers/gpu/nvgpu/os/linux/platform_gv11b_tegra.c
index dafa05e5a..6dde735a2 100644
--- a/drivers/gpu/nvgpu/os/linux/platform_gv11b_tegra.c
+++ b/drivers/gpu/nvgpu/os/linux/platform_gv11b_tegra.c
@@ -276,6 +276,7 @@ struct gk20a_platform gv11b_tegra_platform = {
 
 	.honors_aperture = true,
 	.unified_memory = true,
+	.unify_address_spaces = true,
 	.dma_mask = DMA_BIT_MASK(36),
 
 	.reset_assert = gp10b_tegra_reset_assert,