struct spi_device *spi,
                             struct spi_transfer *transfer)
 {
+       struct dw_spi *dws = spi_controller_get_devdata(master);
        struct chip_data *chip = spi_get_ctldata(spi);
        u32 cr0;
 
        /* CTRLR0[13] Shift Register Loop */
        cr0 |= ((spi->mode & SPI_LOOP) ? 1 : 0) << DWC_SSI_CTRLR0_SRL_OFFSET;
 
+       if (dws->caps & DW_SPI_CAP_KEEMBAY_MST)
+               cr0 |= DWC_SSI_CTRLR0_KEEMBAY_MST;
+
        return cr0;
 }
 EXPORT_SYMBOL_GPL(dw_spi_update_cr0_v1_01a);
 
 #define SPARX5_FORCE_ENA                       0xa4
 #define SPARX5_FORCE_VAL                       0xa8
 
-/*
- * For Keem Bay, CTRLR0[31] is used to select controller mode.
- * 0: SSI is slave
- * 1: SSI is master
- */
-#define KEEMBAY_CTRLR0_SSIC_IS_MST             BIT(31)
-
 struct dw_spi_mscc {
        struct regmap       *syscon;
        void __iomem        *spi_mst; /* Not sparx5 */
        return 0;
 }
 
-static u32 dw_spi_update_cr0_keembay(struct spi_controller *master,
-                                    struct spi_device *spi,
-                                    struct spi_transfer *transfer)
-{
-       u32 cr0 = dw_spi_update_cr0_v1_01a(master, spi, transfer);
-
-       return cr0 | KEEMBAY_CTRLR0_SSIC_IS_MST;
-}
-
 static int dw_spi_keembay_init(struct platform_device *pdev,
                               struct dw_spi_mmio *dwsmmio)
 {
+       dwsmmio->dws.caps = DW_SPI_CAP_KEEMBAY_MST;
+
        /* Register hook to configure CTRLR0 */
-       dwsmmio->dws.update_cr0 = dw_spi_update_cr0_keembay;
+       dwsmmio->dws.update_cr0 = dw_spi_update_cr0_v1_01a;
 
        return 0;
 }
 
 #define DWC_SSI_CTRLR0_FRF_OFFSET      6
 #define DWC_SSI_CTRLR0_DFS_OFFSET      0
 
+/*
+ * For Keem Bay, CTRLR0[31] is used to select controller mode.
+ * 0: SSI is slave
+ * 1: SSI is master
+ */
+#define DWC_SSI_CTRLR0_KEEMBAY_MST     BIT(31)
+
 /* Bit fields in SR, 7 bits */
 #define SR_MASK                                0x7f            /* cover 7 bits */
 #define SR_BUSY                                (1 << 0)
 
 /* DW SPI capabilities */
 #define DW_SPI_CAP_CS_OVERRIDE         BIT(0)
+#define DW_SPI_CAP_KEEMBAY_MST         BIT(1)
 
 struct dw_spi;
 struct dw_spi_dma_ops {