In order to avoid synchronization races between tx and rx path, rely on
mt76_connac_skip_fw_pmctrl putting the chip in sleep mode for mt7921 and
mt7663 devices
Signed-off-by: Lorenzo Bianconi <lorenzo@kernel.org>
Signed-off-by: Felix Fietkau <nbd@nbd.name>
 
        mutex_lock(&dev->pm.mutex);
 
-       if (test_and_set_bit(MT76_STATE_PM, &mphy->state))
+       if (mt76_connac_skip_fw_pmctrl(mphy, &dev->pm))
                goto out;
 
        mt7622_trigger_hif_int(dev, true);
 
        spin_unlock_bh(&pm->wake.lock);
 }
 
+static inline bool
+mt76_connac_skip_fw_pmctrl(struct mt76_phy *phy, struct mt76_connac_pm *pm)
+{
+       bool ret;
+
+       spin_lock_bh(&pm->wake.lock);
+       ret = pm->wake.count || test_and_set_bit(MT76_STATE_PM, &phy->state);
+       spin_unlock_bh(&pm->wake.lock);
+
+       return ret;
+}
+
 static inline void
 mt76_connac_mutex_acquire(struct mt76_dev *dev, struct mt76_connac_pm *pm)
        __acquires(&dev->mutex)
 
 
        mutex_lock(&dev->pm.mutex);
 
-       if (test_and_set_bit(MT76_STATE_PM, &mphy->state))
+       if (mt76_connac_skip_fw_pmctrl(mphy, &dev->pm))
                goto out;
 
        for (i = 0; i < MT7921_DRV_OWN_RETRY_COUNT; i++) {