void mt76x02_mac_poll_tx_status(struct mt76x02_dev *dev, bool irq)
 {
        struct mt76x02_tx_status stat = {};
-       unsigned long flags;
        u8 update = 1;
        bool ret;
 
        trace_mac_txstat_poll(dev);
 
        while (!irq || !kfifo_is_full(&dev->txstatus_fifo)) {
-               spin_lock_irqsave(&dev->mt76.mmio.irq_lock, flags);
+               if (!spin_trylock(&dev->txstatus_fifo_lock))
+                       break;
+
                ret = mt76x02_mac_load_tx_status(dev, &stat);
-               spin_unlock_irqrestore(&dev->mt76.mmio.irq_lock, flags);
+               spin_unlock(&dev->txstatus_fifo_lock);
 
                if (!ret)
                        break;
 
        tasklet_init(&dev->pre_tbtt_tasklet, mt76x02_pre_tbtt_tasklet,
                     (unsigned long)dev);
 
+       spin_lock_init(&dev->txstatus_fifo_lock);
        kfifo_init(&dev->txstatus_fifo, status_fifo, fifo_size);
 
        mt76_dma_attach(&dev->mt76);