struct rtq9128_data {
        struct gpio_desc *enable;
+       unsigned int daifmt;
        int tdm_slots;
        int tdm_slot_width;
        bool tdm_input_data2_select;
 static int rtq9128_dai_set_fmt(struct snd_soc_dai *dai, unsigned int fmt)
 {
        struct rtq9128_data *data = snd_soc_dai_get_drvdata(dai);
-       struct snd_soc_component *comp = dai->component;
        struct device *dev = dai->dev;
-       unsigned int audfmt, fmtval;
-       int ret;
 
        dev_dbg(dev, "%s: fmt 0x%8x\n", __func__, fmt);
 
                return -EINVAL;
        }
 
-       fmtval = fmt & SND_SOC_DAIFMT_FORMAT_MASK;
-       if (data->tdm_slots && fmtval != SND_SOC_DAIFMT_DSP_A && fmtval != SND_SOC_DAIFMT_DSP_B) {
-               dev_err(dev, "TDM is used, format only support DSP_A or DSP_B\n");
-               return -EINVAL;
-       }
+       /* Store here and will be used in runtime hw_params for DAI format setting */
+       data->daifmt = fmt;
 
-       switch (fmtval) {
-       case SND_SOC_DAIFMT_I2S:
-               audfmt = 8;
-               break;
-       case SND_SOC_DAIFMT_LEFT_J:
-               audfmt = 9;
-               break;
-       case SND_SOC_DAIFMT_RIGHT_J:
-               audfmt = 10;
-               break;
-       case SND_SOC_DAIFMT_DSP_A:
-               audfmt = data->tdm_slots ? 12 : 11;
-               break;
-       case SND_SOC_DAIFMT_DSP_B:
-               audfmt = data->tdm_slots ? 4 : 3;
-               break;
-       default:
-               dev_err(dev, "Unsupported format 0x%8x\n", fmt);
-               return -EINVAL;
-       }
-
-       ret = snd_soc_component_write_field(comp, RTQ9128_REG_I2S_OPT, RTQ9128_AUDFMT_MASK, audfmt);
-       return ret < 0 ? ret : 0;
+       return 0;
 }
 
 static int rtq9128_dai_set_tdm_slot(struct snd_soc_dai *dai, unsigned int tx_mask,
        unsigned int width, slot_width, bitrate, audbit, dolen;
        struct snd_soc_component *comp = dai->component;
        struct device *dev = dai->dev;
+       unsigned int fmtval, audfmt;
        int ret;
 
        dev_dbg(dev, "%s: width %d\n", __func__, params_width(param));
 
+       fmtval = FIELD_GET(SND_SOC_DAIFMT_FORMAT_MASK, data->daifmt);
+       if (data->tdm_slots && fmtval != SND_SOC_DAIFMT_DSP_A && fmtval != SND_SOC_DAIFMT_DSP_B) {
+               dev_err(dev, "TDM is used, format only support DSP_A or DSP_B\n");
+               return -EINVAL;
+       }
+
+       switch (fmtval) {
+       case SND_SOC_DAIFMT_I2S:
+               audfmt = 8;
+               break;
+       case SND_SOC_DAIFMT_LEFT_J:
+               audfmt = 9;
+               break;
+       case SND_SOC_DAIFMT_RIGHT_J:
+               audfmt = 10;
+               break;
+       case SND_SOC_DAIFMT_DSP_A:
+               audfmt = data->tdm_slots ? 12 : 11;
+               break;
+       case SND_SOC_DAIFMT_DSP_B:
+               audfmt = data->tdm_slots ? 4 : 3;
+               break;
+       default:
+               dev_err(dev, "Unsupported format 0x%8x\n", fmtval);
+               return -EINVAL;
+       }
+
        switch (width = params_width(param)) {
        case 16:
                audbit = 0;
                return -EINVAL;
        }
 
+       ret = snd_soc_component_write_field(comp, RTQ9128_REG_I2S_OPT, RTQ9128_AUDFMT_MASK, audfmt);
+       if (ret < 0)
+               return ret;
+
        ret = snd_soc_component_write_field(comp, RTQ9128_REG_I2S_OPT, RTQ9128_AUDBIT_MASK, audbit);
        if (ret < 0)
                return ret;