diff --git a/drivers/video/tegra/host/nvdla/nvdla.c b/drivers/video/tegra/host/nvdla/nvdla.c
index e53152b9..ed887d6a 100644
--- a/drivers/video/tegra/host/nvdla/nvdla.c
+++ b/drivers/video/tegra/host/nvdla/nvdla.c
@@ -714,7 +714,7 @@ static struct of_device_id tegra_nvdla_of_match[] = {
 
 static int nvdla_probe(struct platform_device *pdev)
 {
-	int err = 0;
+	int err = 0, dla_dma_bit_mask = 39;
 	struct nvhost_device_data *pdata = NULL;
 	struct nvdla_device *nvdla_dev = NULL;
 	struct device *dev = &pdev->dev;
@@ -759,7 +759,10 @@ static int nvdla_probe(struct platform_device *pdev)
 		goto err_no_ip;
 	}
 
-	dma_set_mask(dev, DMA_BIT_MASK(39));
+	if (tegra_platform_is_vdk())
+		dla_dma_bit_mask = 32;
+
+	dma_set_mask(dev, DMA_BIT_MASK(dla_dma_bit_mask));
 
 	nvdla_dev = devm_kzalloc(dev, sizeof(*nvdla_dev), GFP_KERNEL);
 	if (!nvdla_dev) {