wifi: mt76: mt7915: Fix VHT beamforming capabilities with DBDC

Do not hardcode beamformee 4 space-time streams support, because
DBDC can only support up to 2.

Fixes: 00b2e16e0063 ("mt76: mt7915: add TxBF capabilities")
Signed-off-by: Nicolas Cavallari <nicolas.cavallari@green-communications.fr>
Signed-off-by: Felix Fietkau <nbd@nbd.name>

diff --git a/drivers/net/wireless/mediatek/mt76/mt7915/init.c b/drivers/net/wireless/mediatek/mt76/mt7915/init.c
index e2e43a96416ed82372b314e38029f7c453edd05f..dbda7496bb5e70e6445b8c10134cc64d5ba150e9 100644 (file)
--- a/drivers/net/wireless/mediatek/mt76/mt7915/init.c
+++ b/drivers/net/wireless/mediatek/mt76/mt7915/init.c
@@ -735,7 +735,8 @@ void mt7915_set_stream_vht_txbf_caps(struct mt7915_phy *phy)
 
        *cap |= IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE |
                IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE |
-               (3 << IEEE80211_VHT_CAP_BEAMFORMEE_STS_SHIFT);
+               FIELD_PREP(IEEE80211_VHT_CAP_BEAMFORMEE_STS_MASK,
+                          sts - 1);
 
        *cap &= ~(IEEE80211_VHT_CAP_SOUNDING_DIMENSIONS_MASK |
                  IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE |