dev->hw->wiphy->bands[band] = NULL;
 }
 
+struct mt76_dev *
+mt76_alloc_device(unsigned int size, const struct ieee80211_ops *ops)
+{
+       struct ieee80211_hw *hw;
+       struct mt76_dev *dev;
+
+       hw = ieee80211_alloc_hw(size, ops);
+       if (!hw)
+               return NULL;
+
+       dev = hw->priv;
+       dev->hw = hw;
+       spin_lock_init(&dev->rx_lock);
+       spin_lock_init(&dev->lock);
+       spin_lock_init(&dev->cc_lock);
+
+       return dev;
+}
+EXPORT_SYMBOL_GPL(mt76_alloc_device);
+
 int mt76_register_device(struct mt76_dev *dev, bool vht,
                         struct ieee80211_rate *rates, int n_rates)
 {
 
        dev_set_drvdata(dev->dev, dev);
 
-       spin_lock_init(&dev->lock);
-       spin_lock_init(&dev->cc_lock);
        INIT_LIST_HEAD(&dev->txwi_cache);
 
        SET_IEEE80211_DEV(hw, dev->dev);
 
        return &msband->chan[idx];
 }
 
+struct mt76_dev *mt76_alloc_device(unsigned int size,
+                                  const struct ieee80211_ops *ops);
 int mt76_register_device(struct mt76_dev *dev, bool vht,
                         struct ieee80211_rate *rates, int n_rates);
 void mt76_unregister_device(struct mt76_dev *dev);
 
                .rx_poll_complete = mt76x2_rx_poll_complete,
                .sta_ps = mt76x2_sta_ps,
        };
-       struct ieee80211_hw *hw;
        struct mt76x2_dev *dev;
+       struct mt76_dev *mdev;
 
-       hw = ieee80211_alloc_hw(sizeof(*dev), &mt76x2_ops);
-       if (!hw)
+       mdev = mt76_alloc_device(sizeof(*dev), &mt76x2_ops);
+       if (!mdev)
                return NULL;
 
-       dev = hw->priv;
-       dev->mt76.dev = pdev;
-       dev->mt76.hw = hw;
-       dev->mt76.drv = &drv_ops;
+       dev = container_of(mdev, struct mt76x2_dev, mt76);
+       mdev->dev = pdev;
+       mdev->drv = &drv_ops;
        mutex_init(&dev->mutex);
        spin_lock_init(&dev->irq_lock);
-       spin_lock_init(&dev->mt76.rx_lock);
 
        return dev;
 }