switch (cpu_dai->id) {
        case WSA_CODEC_DMA_RX_0:
+       case RX_CODEC_DMA_RX_0:
+       case RX_CODEC_DMA_RX_1:
+       case TX_CODEC_DMA_TX_0:
+       case TX_CODEC_DMA_TX_1:
+       case TX_CODEC_DMA_TX_2:
+       case TX_CODEC_DMA_TX_3:
                for_each_rtd_codec_dais(rtd, i, codec_dai) {
                        sruntime = snd_soc_dai_get_sdw_stream(codec_dai,
                                                      substream->stream);
        switch (cpu_dai->id) {
        case WSA_CODEC_DMA_RX_0:
        case WSA_CODEC_DMA_RX_1:
+       case RX_CODEC_DMA_RX_0:
+       case RX_CODEC_DMA_RX_1:
+       case TX_CODEC_DMA_TX_0:
+       case TX_CODEC_DMA_TX_1:
+       case TX_CODEC_DMA_TX_2:
+       case TX_CODEC_DMA_TX_3:
                return sm8250_snd_wsa_dma_prepare(substream);
        default:
                break;
        switch (cpu_dai->id) {
        case WSA_CODEC_DMA_RX_0:
        case WSA_CODEC_DMA_RX_1:
+       case RX_CODEC_DMA_RX_0:
+       case RX_CODEC_DMA_RX_1:
+       case TX_CODEC_DMA_TX_0:
+       case TX_CODEC_DMA_TX_1:
+       case TX_CODEC_DMA_TX_2:
+       case TX_CODEC_DMA_TX_3:
                if (sruntime && data->stream_prepared[cpu_dai->id]) {
                        sdw_disable_stream(sruntime);
                        sdw_deprepare_stream(sruntime);