#include <linux/platform_device.h>
 #include <linux/delay.h>
 #include <linux/spi/spi.h>
+#include <linux/iopoll.h>
 
 #define AMD_SPI_CTRL0_REG      0x00
 #define AMD_SPI_EXEC_CMD       BIT(16)
 
 static int amd_spi_busy_wait(struct amd_spi *amd_spi)
 {
-       int timeout = 100000;
+       u32 val;
 
-       /* poll for SPI bus to become idle */
-       while (amd_spi_readreg32(amd_spi, AMD_SPI_CTRL0_REG) & AMD_SPI_BUSY) {
-               usleep_range(10, 20);
-               if (timeout-- < 0)
-                       return -ETIMEDOUT;
-       }
-
-       return 0;
+       return readl_poll_timeout(amd_spi->io_remap_addr + AMD_SPI_CTRL0_REG,
+                                 val, !(val & AMD_SPI_BUSY), 20, 2000000);
 }
 
 static int amd_spi_execute_opcode(struct amd_spi *amd_spi)