DECLARE_EWMA(signal, 10, 8);
 
+#define MT_WCID_TX_INFO_RATE           GENMASK(15, 0)
+#define MT_WCID_TX_INFO_NSS            GENMASK(17, 16)
+#define MT_WCID_TX_INFO_TXPWR_ADJ      GENMASK(25, 18)
+#define MT_WCID_TX_INFO_SET            BIT(31)
+
 struct mt76_wcid {
        struct mt76_rx_tid __rcu *aggr[IEEE80211_NUM_TIDS];
 
        u8 rx_check_pn;
        u8 rx_key_pn[IEEE80211_NUM_TIDS][6];
 
-       __le16 tx_rate;
-       bool tx_rate_set;
-       u8 tx_rate_nss;
-       s8 max_txpwr_adj;
+       u32 tx_info;
        bool sw_iv;
 
        u8 packet_id;
 
                MT_WTBL_UPDATE_RATE_UPDATE |
                MT_WTBL_UPDATE_TX_COUNT_CLEAR);
 
-       if (!sta->wcid.tx_rate_set)
+       if (!(sta->wcid.tx_info & MT_WCID_TX_INFO_SET))
                mt76_poll(dev, MT_WTBL_UPDATE, MT_WTBL_UPDATE_BUSY, 0, 5000);
 
        sta->rate_count = 2 * MT7603_RATE_RETRY * n_rates;
-       sta->wcid.tx_rate_set = true;
+       sta->wcid.tx_info |= MT_WCID_TX_INFO_SET;
 }
 
 static enum mt7603_cipher_type
 
 void mt76x02_mac_wcid_set_rate(struct mt76x02_dev *dev, struct mt76_wcid *wcid,
                               const struct ieee80211_tx_rate *rate)
 {
-       spin_lock_bh(&dev->mt76.lock);
-       wcid->tx_rate = mt76x02_mac_tx_rate_val(dev, rate, &wcid->tx_rate_nss);
-       wcid->tx_rate_set = true;
-       spin_unlock_bh(&dev->mt76.lock);
+       s8 max_txpwr_adj = mt76x02_tx_get_max_txpwr_adj(dev, rate);
+       __le16 rateval;
+       u32 tx_info;
+       s8 nss;
+
+       rateval = mt76x02_mac_tx_rate_val(dev, rate, &nss);
+       tx_info = FIELD_PREP(MT_WCID_TX_INFO_RATE, rateval) |
+                 FIELD_PREP(MT_WCID_TX_INFO_NSS, nss) |
+                 FIELD_PREP(MT_WCID_TX_INFO_TXPWR_ADJ, max_txpwr_adj) |
+                 MT_WCID_TX_INFO_SET;
+       wcid->tx_info = tx_info;
 }
 
 void mt76x02_mac_set_short_preamble(struct mt76x02_dev *dev, bool enable)
        struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
        struct ieee80211_tx_rate *rate = &info->control.rates[0];
        struct ieee80211_key_conf *key = info->control.hw_key;
+       u32 wcid_tx_info;
        u16 rate_ht_mask = FIELD_PREP(MT_RXWI_RATE_PHY, BIT(1) | BIT(2));
        u16 txwi_flags = 0;
        u8 nss;
                txwi->eiv = *((__le32 *)&ccmp_pn[4]);
        }
 
-       spin_lock_bh(&dev->mt76.lock);
        if (wcid && (rate->idx < 0 || !rate->count)) {
-               txwi->rate = wcid->tx_rate;
-               max_txpwr_adj = wcid->max_txpwr_adj;
-               nss = wcid->tx_rate_nss;
+               wcid_tx_info = wcid->tx_info;
+               txwi->rate = FIELD_GET(MT_WCID_TX_INFO_RATE, wcid_tx_info);
+               max_txpwr_adj = FIELD_GET(MT_WCID_TX_INFO_TXPWR_ADJ,
+                                         wcid_tx_info);
+               nss = FIELD_GET(MT_WCID_TX_INFO_NSS, wcid_tx_info);
        } else {
                txwi->rate = mt76x02_mac_tx_rate_val(dev, rate, &nss);
                max_txpwr_adj = mt76x02_tx_get_max_txpwr_adj(dev, rate);
        }
-       spin_unlock_bh(&dev->mt76.lock);
 
        txpwr_adj = mt76x02_tx_get_txpwr_adj(dev, dev->mt76.txpower_conf,
                                             max_txpwr_adj);
 
                skb_set_queue_mapping(skb, qid);
        }
 
-       if (!wcid->tx_rate_set)
+       if (!(wcid->tx_info & MT_WCID_TX_INFO_SET))
                ieee80211_get_tx_rates(info->control.vif, sta, skb,
                                       info->control.rates, 1);
 
        }
 
        info = IEEE80211_SKB_CB(skb);
-       if (!wcid->tx_rate_set)
+       if (!(wcid->tx_info & MT_WCID_TX_INFO_SET))
                ieee80211_get_tx_rates(txq->vif, txq->sta, skb,
                                       info->control.rates, 1);
        tx_rate = info->control.rates[0];