EXPORT_SYMBOL_GPL(mt76x02_rates);
 
 void mt76x02_configure_filter(struct ieee80211_hw *hw,
-                            unsigned int changed_flags,
-                            unsigned int *total_flags, u64 multicast)
+                             unsigned int changed_flags,
+                             unsigned int *total_flags, u64 multicast)
 {
-       struct mt76_dev *dev = hw->priv;
+       struct mt76x02_dev *dev = hw->priv;
        u32 flags = 0;
 
 #define MT76_FILTER(_flag, _hw) do { \
                flags |= *total_flags & FIF_##_flag;                    \
-               dev->rxfilter &= ~(_hw);                                \
-               dev->rxfilter |= !(flags & FIF_##_flag) * (_hw);        \
+               dev->mt76.rxfilter &= ~(_hw);                           \
+               dev->mt76.rxfilter |= !(flags & FIF_##_flag) * (_hw);   \
        } while (0)
 
-       mutex_lock(&dev->mutex);
+       mutex_lock(&dev->mt76.mutex);
 
-       dev->rxfilter &= ~MT_RX_FILTR_CFG_OTHER_BSS;
+       dev->mt76.rxfilter &= ~MT_RX_FILTR_CFG_OTHER_BSS;
 
        MT76_FILTER(FCSFAIL, MT_RX_FILTR_CFG_CRC_ERR);
        MT76_FILTER(PLCPFAIL, MT_RX_FILTR_CFG_PHY_ERR);
        MT76_FILTER(PSPOLL, MT_RX_FILTR_CFG_PSPOLL);
 
        *total_flags = flags;
-       dev->bus->wr(dev, MT_RX_FILTR_CFG, dev->rxfilter);
+       mt76_wr(dev, MT_RX_FILTR_CFG, dev->mt76.rxfilter);
 
-       mutex_unlock(&dev->mutex);
+       mutex_unlock(&dev->mt76.mutex);
 }
 EXPORT_SYMBOL_GPL(mt76x02_configure_filter);
 
 int mt76x02_sta_add(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
-                  struct ieee80211_sta *sta)
+                   struct ieee80211_sta *sta)
 {
-       struct mt76_dev *dev = hw->priv;
-       struct mt76x02_sta *msta = (struct mt76x02_sta *) sta->drv_priv;
-       struct mt76x02_vif *mvif = (struct mt76x02_vif *) vif->drv_priv;
+       struct mt76x02_dev *dev = hw->priv;
+       struct mt76x02_sta *msta = (struct mt76x02_sta *)sta->drv_priv;
+       struct mt76x02_vif *mvif = (struct mt76x02_vif *)vif->drv_priv;
        int ret = 0;
        int idx = 0;
        int i;
 
-       mutex_lock(&dev->mutex);
+       mutex_lock(&dev->mt76.mutex);
 
-       idx = mt76_wcid_alloc(dev->wcid_mask, ARRAY_SIZE(dev->wcid));
+       idx = mt76_wcid_alloc(dev->mt76.wcid_mask, ARRAY_SIZE(dev->mt76.wcid));
        if (idx < 0) {
                ret = -ENOSPC;
                goto out;
        msta->wcid.sta = 1;
        msta->wcid.idx = idx;
        msta->wcid.hw_key_idx = -1;
-       mt76x02_mac_wcid_setup(dev, idx, mvif->idx, sta->addr);
-       mt76x02_mac_wcid_set_drop(dev, idx, false);
+       mt76x02_mac_wcid_setup(&dev->mt76, idx, mvif->idx, sta->addr);
+       mt76x02_mac_wcid_set_drop(&dev->mt76, idx, false);
        for (i = 0; i < ARRAY_SIZE(sta->txq); i++)
-               mt76x02_txq_init(dev, sta->txq[i]);
+               mt76x02_txq_init(&dev->mt76, sta->txq[i]);
 
        if (vif->type == NL80211_IFTYPE_AP)
                set_bit(MT_WCID_FLAG_CHECK_PS, &msta->wcid.flags);
 
        ewma_signal_init(&msta->rssi);
 
-       rcu_assign_pointer(dev->wcid[idx], &msta->wcid);
+       rcu_assign_pointer(dev->mt76.wcid[idx], &msta->wcid);
 
 out:
-       mutex_unlock(&dev->mutex);
+       mutex_unlock(&dev->mt76.mutex);
 
        return ret;
 }
 EXPORT_SYMBOL_GPL(mt76x02_sta_add);
 
 int mt76x02_sta_remove(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
-                     struct ieee80211_sta *sta)
+                      struct ieee80211_sta *sta)
 {
-       struct mt76_dev *dev = hw->priv;
-       struct mt76x02_sta *msta = (struct mt76x02_sta *) sta->drv_priv;
+       struct mt76x02_dev *dev = hw->priv;
+       struct mt76x02_sta *msta = (struct mt76x02_sta *)sta->drv_priv;
        int idx = msta->wcid.idx;
        int i;
 
-       mutex_lock(&dev->mutex);
-       rcu_assign_pointer(dev->wcid[idx], NULL);
+       mutex_lock(&dev->mt76.mutex);
+       rcu_assign_pointer(dev->mt76.wcid[idx], NULL);
        for (i = 0; i < ARRAY_SIZE(sta->txq); i++)
-               mt76_txq_remove(dev, sta->txq[i]);
-       mt76x02_mac_wcid_set_drop(dev, idx, true);
-       mt76_wcid_free(dev->wcid_mask, idx);
-       mt76x02_mac_wcid_setup(dev, idx, 0, NULL);
-       mutex_unlock(&dev->mutex);
+               mt76_txq_remove(&dev->mt76, sta->txq[i]);
+       mt76x02_mac_wcid_set_drop(&dev->mt76, idx, true);
+       mt76_wcid_free(dev->mt76.wcid_mask, idx);
+       mt76x02_mac_wcid_setup(&dev->mt76, idx, 0, NULL);
+       mutex_unlock(&dev->mt76.mutex);
 
        return 0;
 }
 EXPORT_SYMBOL_GPL(mt76x02_sta_remove);
 
-void mt76x02_vif_init(struct mt76_dev *dev, struct ieee80211_vif *vif,
-                    unsigned int idx)
+void mt76x02_vif_init(struct mt76x02_dev *dev, struct ieee80211_vif *vif,
+                     unsigned int idx)
 {
-       struct mt76x02_vif *mvif = (struct mt76x02_vif *) vif->drv_priv;
+       struct mt76x02_vif *mvif = (struct mt76x02_vif *)vif->drv_priv;
 
        mvif->idx = idx;
        mvif->group_wcid.idx = MT_VIF_WCID(idx);
        mvif->group_wcid.hw_key_idx = -1;
-       mt76x02_txq_init(dev, vif->txq);
+       mt76x02_txq_init(&dev->mt76, vif->txq);
 }
 EXPORT_SYMBOL_GPL(mt76x02_vif_init);
 
 int
 mt76x02_add_interface(struct ieee80211_hw *hw, struct ieee80211_vif *vif)
 {
-       struct mt76_dev *dev = hw->priv;
+       struct mt76x02_dev *dev = hw->priv;
        unsigned int idx = 0;
 
        if (vif->addr[0] & BIT(1))
-               idx = 1 + (((dev->macaddr[0] ^ vif->addr[0]) >> 2) & 7);
+               idx = 1 + (((dev->mt76.macaddr[0] ^ vif->addr[0]) >> 2) & 7);
 
        /*
         * Client mode typically only has one configurable BSSID register,
 EXPORT_SYMBOL_GPL(mt76x02_add_interface);
 
 void mt76x02_remove_interface(struct ieee80211_hw *hw,
-                            struct ieee80211_vif *vif)
+                             struct ieee80211_vif *vif)
 {
-       struct mt76_dev *dev = hw->priv;
+       struct mt76x02_dev *dev = hw->priv;
 
-       mt76_txq_remove(dev, vif->txq);
+       mt76_txq_remove(&dev->mt76, vif->txq);
 }
 EXPORT_SYMBOL_GPL(mt76x02_remove_interface);
 
 int mt76x02_ampdu_action(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
-                       struct ieee80211_ampdu_params *params)
+                        struct ieee80211_ampdu_params *params)
 {
        enum ieee80211_ampdu_mlme_action action = params->action;
        struct ieee80211_sta *sta = params->sta;
-       struct mt76_dev *dev = hw->priv;
+       struct mt76x02_dev *dev = hw->priv;
        struct mt76x02_sta *msta = (struct mt76x02_sta *) sta->drv_priv;
        struct ieee80211_txq *txq = sta->txq[params->tid];
        u16 tid = params->tid;
 
        switch (action) {
        case IEEE80211_AMPDU_RX_START:
-               mt76_rx_aggr_start(dev, &msta->wcid, tid, *ssn, params->buf_size);
-               __mt76_set(dev, MT_WCID_ADDR(msta->wcid.idx) + 4, BIT(16 + tid));
+               mt76_rx_aggr_start(&dev->mt76, &msta->wcid, tid,
+                                  *ssn, params->buf_size);
+               mt76_set(dev, MT_WCID_ADDR(msta->wcid.idx) + 4, BIT(16 + tid));
                break;
        case IEEE80211_AMPDU_RX_STOP:
-               mt76_rx_aggr_stop(dev, &msta->wcid, tid);
-               __mt76_clear(dev, MT_WCID_ADDR(msta->wcid.idx) + 4, BIT(16 + tid));
+               mt76_rx_aggr_stop(&dev->mt76, &msta->wcid, tid);
+               mt76_clear(dev, MT_WCID_ADDR(msta->wcid.idx) + 4,
+                          BIT(16 + tid));
                break;
        case IEEE80211_AMPDU_TX_OPERATIONAL:
                mtxq->aggr = true;
 EXPORT_SYMBOL_GPL(mt76x02_ampdu_action);
 
 int mt76x02_set_key(struct ieee80211_hw *hw, enum set_key_cmd cmd,
-                  struct ieee80211_vif *vif, struct ieee80211_sta *sta,
-                  struct ieee80211_key_conf *key)
+                   struct ieee80211_vif *vif, struct ieee80211_sta *sta,
+                   struct ieee80211_key_conf *key)
 {
-       struct mt76_dev *dev = hw->priv;
-       struct mt76x02_vif *mvif = (struct mt76x02_vif *) vif->drv_priv;
+       struct mt76x02_dev *dev = hw->priv;
+       struct mt76x02_vif *mvif = (struct mt76x02_vif *)vif->drv_priv;
        struct mt76x02_sta *msta;
        struct mt76_wcid *wcid;
        int idx = key->keyidx;
 
                key = NULL;
        }
-       mt76_wcid_key_setup(dev, wcid, key);
+       mt76_wcid_key_setup(&dev->mt76, wcid, key);
 
        if (!msta) {
                if (key || wcid->hw_key_idx == idx) {
-                       ret = mt76x02_mac_wcid_set_key(dev, wcid->idx, key);
+                       ret = mt76x02_mac_wcid_set_key(&dev->mt76,
+                                                      wcid->idx, key);
                        if (ret)
                                return ret;
                }
 
-               return mt76x02_mac_shared_key_setup(dev, mvif->idx, idx, key);
+               return mt76x02_mac_shared_key_setup(&dev->mt76,
+                                                   mvif->idx, idx, key);
        }
 
-       return mt76x02_mac_wcid_set_key(dev, msta->wcid.idx, key);
+       return mt76x02_mac_wcid_set_key(&dev->mt76, msta->wcid.idx, key);
 }
 EXPORT_SYMBOL_GPL(mt76x02_set_key);
 
 int mt76x02_conf_tx(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
-                  u16 queue, const struct ieee80211_tx_queue_params *params)
+                   u16 queue, const struct ieee80211_tx_queue_params *params)
 {
-       struct mt76_dev *dev = hw->priv;
+       struct mt76x02_dev *dev = hw->priv;
        u8 cw_min = 5, cw_max = 10, qid;
        u32 val;
 
-       qid = dev->q_tx[queue].hw_idx;
+       qid = dev->mt76.q_tx[queue].hw_idx;
 
        if (params->cw_min)
                cw_min = fls(params->cw_min);
              FIELD_PREP(MT_EDCA_CFG_AIFSN, params->aifs) |
              FIELD_PREP(MT_EDCA_CFG_CWMIN, cw_min) |
              FIELD_PREP(MT_EDCA_CFG_CWMAX, cw_max);
-       __mt76_wr(dev, MT_EDCA_CFG_AC(qid), val);
+       mt76_wr(dev, MT_EDCA_CFG_AC(qid), val);
 
-       val = __mt76_rr(dev, MT_WMM_TXOP(qid));
+       val = mt76_rr(dev, MT_WMM_TXOP(qid));
        val &= ~(MT_WMM_TXOP_MASK << MT_WMM_TXOP_SHIFT(qid));
        val |= params->txop << MT_WMM_TXOP_SHIFT(qid);
-       __mt76_wr(dev, MT_WMM_TXOP(qid), val);
+       mt76_wr(dev, MT_WMM_TXOP(qid), val);
 
-       val = __mt76_rr(dev, MT_WMM_AIFSN);
+       val = mt76_rr(dev, MT_WMM_AIFSN);
        val &= ~(MT_WMM_AIFSN_MASK << MT_WMM_AIFSN_SHIFT(qid));
        val |= params->aifs << MT_WMM_AIFSN_SHIFT(qid);
-       __mt76_wr(dev, MT_WMM_AIFSN, val);
+       mt76_wr(dev, MT_WMM_AIFSN, val);
 
-       val = __mt76_rr(dev, MT_WMM_CWMIN);
+       val = mt76_rr(dev, MT_WMM_CWMIN);
        val &= ~(MT_WMM_CWMIN_MASK << MT_WMM_CWMIN_SHIFT(qid));
        val |= cw_min << MT_WMM_CWMIN_SHIFT(qid);
-       __mt76_wr(dev, MT_WMM_CWMIN, val);
+       mt76_wr(dev, MT_WMM_CWMIN, val);
 
-       val = __mt76_rr(dev, MT_WMM_CWMAX);
+       val = mt76_rr(dev, MT_WMM_CWMAX);
        val &= ~(MT_WMM_CWMAX_MASK << MT_WMM_CWMAX_SHIFT(qid));
        val |= cw_max << MT_WMM_CWMAX_SHIFT(qid);
-       __mt76_wr(dev, MT_WMM_CWMAX, val);
+       mt76_wr(dev, MT_WMM_CWMAX, val);
 
        return 0;
 }
 };
 EXPORT_SYMBOL_GPL(mt76x02_beacon_offsets);
 
-void mt76x02_set_beacon_offsets(struct mt76_dev *dev)
+void mt76x02_set_beacon_offsets(struct mt76x02_dev *dev)
 {
        u16 val, base = MT_BEACON_BASE;
        u32 regs[4] = {};
        }
 
        for (i = 0; i < 4; i++)
-               __mt76_wr(dev, MT_BCN_OFFSET(i), regs[i]);
+               mt76_wr(dev, MT_BCN_OFFSET(i), regs[i]);
 }
 EXPORT_SYMBOL_GPL(mt76x02_set_beacon_offsets);