#include "iwl-trans.h"
 #include "iwl-drv.h"
 #include "iwl-fh.h"
+#include <linux/dmapool.h>
 
 struct iwl_trans *iwl_trans_alloc(unsigned int priv_size,
                                  struct device *dev,
        trans->ops = ops;
        trans->num_rx_queues = 1;
 
+       if (trans->trans_cfg->device_family >= IWL_DEVICE_FAMILY_AX210)
+               trans->txqs.bc_tbl_size = sizeof(struct iwl_gen3_bc_tbl);
+       else
+               trans->txqs.bc_tbl_size = sizeof(struct iwlagn_scd_bc_tbl);
+       /*
+        * For gen2 devices, we use a single allocation for each byte-count
+        * table, but they're pretty small (1k) so use a DMA pool that we
+        * allocate here.
+        */
+       if (trans->trans_cfg->gen2) {
+               trans->txqs.bc_pool = dmam_pool_create("iwlwifi:bc", dev,
+                                                      trans->txqs.bc_tbl_size,
+                                                      256, 0);
+               if (!trans->txqs.bc_pool)
+                       return NULL;
+       }
        snprintf(trans->dev_cmd_pool_name, sizeof(trans->dev_cmd_pool_name),
                 "iwl_cmd_pool:%s", dev_name(trans->dev));
        trans->dev_cmd_pool =
 
        unsigned long queue_used[BITS_TO_LONGS(IWL_MAX_TVQM_QUEUES)];
        unsigned long queue_stopped[BITS_TO_LONGS(IWL_MAX_TVQM_QUEUES)];
        struct iwl_txq *txq[IWL_MAX_TVQM_QUEUES];
+       struct dma_pool *bc_pool;
+       size_t bc_tbl_size;
        struct {
                u8 fifo;
                u8 q_id;
 
        u32 scd_base_addr;
        struct iwl_dma_ptr scd_bc_tbls;
        struct iwl_dma_ptr kw;
-       struct dma_pool *bc_pool;
 
        struct iwl_txq *txq_memory;
 
 
 
        init_waitqueue_head(&trans_pcie->sx_waitq);
 
-       /*
-        * For gen2 devices, we use a single allocation for each byte-count
-        * table, but they're pretty small (1k) so use a DMA pool that we
-        * allocate here.
-        */
-       if (cfg_trans->gen2) {
-               size_t bc_tbl_size;
-
-               if (cfg_trans->device_family >= IWL_DEVICE_FAMILY_AX210)
-                       bc_tbl_size = sizeof(struct iwl_gen3_bc_tbl);
-               else
-                       bc_tbl_size = sizeof(struct iwlagn_scd_bc_tbl);
-
-               trans_pcie->bc_pool = dmam_pool_create("iwlwifi:bc", &pdev->dev,
-                                                      bc_tbl_size, 256, 0);
-               if (!trans_pcie->bc_pool)
-                       goto out_no_pci;
-       }
 
        if (trans_pcie->msix_enabled) {
                ret = iwl_pcie_init_msix_handler(pdev, trans_pcie);
 
 
        kfree(txq->entries);
        if (txq->bc_tbl.addr)
-               dma_pool_free(trans_pcie->bc_pool, txq->bc_tbl.addr,
+               dma_pool_free(trans->txqs.bc_pool, txq->bc_tbl.addr,
                              txq->bc_tbl.dma);
        kfree(txq);
 }
                                     struct iwl_txq **intxq, int size,
                                     unsigned int timeout)
 {
-       struct iwl_trans_pcie *trans_pcie = IWL_TRANS_GET_PCIE_TRANS(trans);
        size_t bc_tbl_size, bc_tbl_entries;
        struct iwl_txq *txq;
        int ret;
 
-       if (trans->trans_cfg->device_family >= IWL_DEVICE_FAMILY_AX210) {
-               bc_tbl_size = sizeof(struct iwl_gen3_bc_tbl);
-               bc_tbl_entries = bc_tbl_size / sizeof(u16);
-       } else {
-               bc_tbl_size = sizeof(struct iwlagn_scd_bc_tbl);
-               bc_tbl_entries = bc_tbl_size / sizeof(u16);
-       }
+       WARN_ON(!trans->txqs.bc_tbl_size);
+
+       bc_tbl_size = trans->txqs.bc_tbl_size;
+       bc_tbl_entries = bc_tbl_size / sizeof(u16);
 
        if (WARN_ON(size > bc_tbl_entries))
                return -EINVAL;
        if (!txq)
                return -ENOMEM;
 
-       txq->bc_tbl.addr = dma_pool_alloc(trans_pcie->bc_pool, GFP_KERNEL,
+       txq->bc_tbl.addr = dma_pool_alloc(trans->txqs.bc_pool, GFP_KERNEL,
                                          &txq->bc_tbl.dma);
        if (!txq->bc_tbl.addr) {
                IWL_ERR(trans, "Scheduler BC Table allocation failed\n");