diff --git a/drivers/gpu/nvgpu/os/linux/platform_ga10b_tegra.c b/drivers/gpu/nvgpu/os/linux/platform_ga10b_tegra.c
index d56a9ba1f..19bb0a4b4 100644
--- a/drivers/gpu/nvgpu/os/linux/platform_ga10b_tegra.c
+++ b/drivers/gpu/nvgpu/os/linux/platform_ga10b_tegra.c
@@ -765,7 +765,7 @@ struct gk20a_platform ga10b_tegra_platform = {
 	 * This specifies the maximum contiguous size of a DMA mapping to Linux
 	 * kernel's DMA framework.
 	 * The IOMMU is capable of mapping all of physical memory and hence
-	 * dma_mask is set to memory size (128GB in this case).
+	 * dma_mask is set to memory size (512GB in this case).
 	 * For iGPU, nvgpu executes own dma allocs (e.g. alloc_page()) and
 	 * sg_table construction. No IOMMU mapping is required and so dma_mask
 	 * value is not important.
@@ -773,7 +773,7 @@ struct gk20a_platform ga10b_tegra_platform = {
 	 * significant. In this case, IOMMU bit in GPU physical address is not
 	 * relevant.
 	 */
-	.dma_mask = DMA_BIT_MASK(37),
+	.dma_mask = DMA_BIT_MASK(39),
 
 	.reset_assert = gp10b_tegra_reset_assert,
 	.reset_deassert = gp10b_tegra_reset_deassert,
diff --git a/drivers/gpu/nvgpu/os/linux/platform_gv11b_tegra.c b/drivers/gpu/nvgpu/os/linux/platform_gv11b_tegra.c
index 88d751374..b0ca6538d 100644
--- a/drivers/gpu/nvgpu/os/linux/platform_gv11b_tegra.c
+++ b/drivers/gpu/nvgpu/os/linux/platform_gv11b_tegra.c
@@ -1,7 +1,7 @@
 /*
  * GV11B Tegra Platform Interface
  *
- * Copyright (c) 2016-2021, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2016-2022, NVIDIA CORPORATION.  All rights reserved.
  *
  * This program is free software; you can redistribute it and/or modify it
  * under the terms and conditions of the GNU General Public License,
@@ -307,7 +307,7 @@ struct gk20a_platform gv11b_tegra_platform = {
 
 	.honors_aperture = true,
 	.unified_memory = true,
-	.dma_mask = DMA_BIT_MASK(36),
+	.dma_mask = DMA_BIT_MASK(38),
 
 	.reset_assert = gp10b_tegra_reset_assert,
 	.reset_deassert = gp10b_tegra_reset_deassert,