diff --git a/drivers/gpu/nvgpu/os/linux/vgpu/gv11b/platform_gv11b_vgpu_tegra.c b/drivers/gpu/nvgpu/os/linux/vgpu/gv11b/platform_gv11b_vgpu_tegra.c
index 252ab4876..1da0ca9d9 100644
--- a/drivers/gpu/nvgpu/os/linux/vgpu/gv11b/platform_gv11b_vgpu_tegra.c
+++ b/drivers/gpu/nvgpu/os/linux/vgpu/gv11b/platform_gv11b_vgpu_tegra.c
@@ -105,5 +105,6 @@ struct gk20a_platform gv11b_vgpu_tegra_platform = {
 	.suspend = vgpu_tegra_suspend,
 	.resume = vgpu_tegra_resume,
 
+	.unify_address_spaces = true,
 	.dma_mask = DMA_BIT_MASK(36),
 };
diff --git a/drivers/gpu/nvgpu/os/linux/vgpu/platform_vgpu_tegra.c b/drivers/gpu/nvgpu/os/linux/vgpu/platform_vgpu_tegra.c
index 948323ef6..df3d61833 100644
--- a/drivers/gpu/nvgpu/os/linux/vgpu/platform_vgpu_tegra.c
+++ b/drivers/gpu/nvgpu/os/linux/vgpu/platform_vgpu_tegra.c
@@ -94,4 +94,6 @@ struct gk20a_platform vgpu_tegra_platform = {
 	/* power management callbacks */
 	.suspend = vgpu_tegra_suspend,
 	.resume = vgpu_tegra_resume,
+
+	.unify_address_spaces = true,
 };