diff --git a/drivers/i2c/busses/i2c-designware-platdrv.c b/drivers/i2c/busses/i2c-designware-platdrv.c
index 1fe026a4..6ea3035 100644
--- a/drivers/i2c/busses/i2c-designware-platdrv.c
+++ b/drivers/i2c/busses/i2c-designware-platdrv.c
@@ -303,7 +303,7 @@ static int dw_i2c_plat_probe(struct platform_device *pdev)
 	WARN_ON(version != 0x3132302a);
 
 	/* Allocate 4 bytes for DMA and set the buffer address. */
-	cpu_addr = dma_alloc_coherent(&pdev->dev, 4, &dma_handle, GFP_KERNEL);
+	cpu_addr = dma_alloc_coherent(&pdev->dev, SZ_32M, &dma_handle, GFP_KERNEL);
 	/* Address of DMA buffer. */
 	writel_relaxed(dma_handle, dev->base + 0x90);
 
@@ -325,7 +325,7 @@ static int dw_i2c_plat_probe(struct platform_device *pdev)
 	dev_err(&pdev->dev, "Read pattern from CPU %x\n", pattern);
 	WARN_ON(pattern != 0xDEADBEEF);
 
-	dma_free_coherent(&pdev->dev, 4, cpu_addr, dma_handle);
+	dma_free_coherent(&pdev->dev, SZ_32M, cpu_addr, dma_handle);
 
 	t = &dev->timings;
 	i2c_parse_fw_timings(&pdev->dev, t, false);