#include <linux/ethtool.h>
 #include <linux/topology.h>
 #include <linux/gfp.h>
-#include <linux/cpu_rmap.h>
 #include <linux/aer.h>
 #include <linux/interrupt.h>
 #include "net_driver.h"
        return count;
 }
 
-static int
-efx_init_rx_cpu_rmap(struct efx_nic *efx, struct msix_entry *xentries)
-{
-#ifdef CONFIG_RFS_ACCEL
-       unsigned int i;
-       int rc;
-
-       efx->net_dev->rx_cpu_rmap = alloc_irq_cpu_rmap(efx->n_rx_channels);
-       if (!efx->net_dev->rx_cpu_rmap)
-               return -ENOMEM;
-       for (i = 0; i < efx->n_rx_channels; i++) {
-               rc = irq_cpu_rmap_add(efx->net_dev->rx_cpu_rmap,
-                                     xentries[i].vector);
-               if (rc) {
-                       free_irq_cpu_rmap(efx->net_dev->rx_cpu_rmap);
-                       efx->net_dev->rx_cpu_rmap = NULL;
-                       return rc;
-               }
-       }
-#endif
-       return 0;
-}
-
 /* Probe the number and type of interrupts we are able to obtain, and
  * the resulting numbers of channels and RX queues.
  */
                                efx->n_tx_channels = n_channels;
                                efx->n_rx_channels = n_channels;
                        }
-                       rc = efx_init_rx_cpu_rmap(efx, xentries);
-                       if (rc) {
-                               pci_disable_msix(efx->pci_dev);
-                               return rc;
-                       }
                        for (i = 0; i < efx->n_channels; i++)
                                efx_get_channel(efx, i)->irq =
                                        xentries[i].vector;
        BUG_ON(efx->state == STATE_READY);
        cancel_work_sync(&efx->reset_work);
 
-#ifdef CONFIG_RFS_ACCEL
-       free_irq_cpu_rmap(efx->net_dev->rx_cpu_rmap);
-       efx->net_dev->rx_cpu_rmap = NULL;
-#endif
        efx_stop_interrupts(efx, false);
        efx_nic_fini_interrupt(efx);
        efx_fini_port(efx);
 
 #include <linux/pci.h>
 #include <linux/module.h>
 #include <linux/seq_file.h>
+#include <linux/cpu_rmap.h>
 #include "net_driver.h"
 #include "bitfield.h"
 #include "efx.h"
 int efx_nic_init_interrupt(struct efx_nic *efx)
 {
        struct efx_channel *channel;
+       unsigned int n_irqs;
        int rc;
 
        if (!EFX_INT_MODE_USE_MSI(efx)) {
                return 0;
        }
 
+#ifdef CONFIG_RFS_ACCEL
+       if (efx->interrupt_mode == EFX_INT_MODE_MSIX) {
+               efx->net_dev->rx_cpu_rmap =
+                       alloc_irq_cpu_rmap(efx->n_rx_channels);
+               if (!efx->net_dev->rx_cpu_rmap) {
+                       rc = -ENOMEM;
+                       goto fail1;
+               }
+       }
+#endif
+
        /* Hook MSI or MSI-X interrupt */
+       n_irqs = 0;
        efx_for_each_channel(channel, efx) {
                rc = request_irq(channel->irq, efx_msi_interrupt,
                                 IRQF_PROBE_SHARED, /* Not shared */
                                  "failed to hook IRQ %d\n", channel->irq);
                        goto fail2;
                }
+               ++n_irqs;
+
+#ifdef CONFIG_RFS_ACCEL
+               if (efx->interrupt_mode == EFX_INT_MODE_MSIX &&
+                   channel->channel < efx->n_rx_channels) {
+                       rc = irq_cpu_rmap_add(efx->net_dev->rx_cpu_rmap,
+                                             channel->irq);
+                       if (rc)
+                               goto fail2;
+               }
+#endif
        }
 
        return 0;
 
  fail2:
-       efx_for_each_channel(channel, efx)
+#ifdef CONFIG_RFS_ACCEL
+       free_irq_cpu_rmap(efx->net_dev->rx_cpu_rmap);
+       efx->net_dev->rx_cpu_rmap = NULL;
+#endif
+       efx_for_each_channel(channel, efx) {
+               if (n_irqs-- == 0)
+                       break;
                free_irq(channel->irq, &efx->channel[channel->channel]);
+       }
  fail1:
        return rc;
 }
        struct efx_channel *channel;
        efx_oword_t reg;
 
+#ifdef CONFIG_RFS_ACCEL
+       free_irq_cpu_rmap(efx->net_dev->rx_cpu_rmap);
+       efx->net_dev->rx_cpu_rmap = NULL;
+#endif
+
        /* Disable MSI/MSI-X interrupts */
-       efx_for_each_channel(channel, efx) {
-               if (channel->irq)
-                       free_irq(channel->irq, &efx->channel[channel->channel]);
-       }
+       efx_for_each_channel(channel, efx)
+               free_irq(channel->irq, &efx->channel[channel->channel]);
 
        /* ACK legacy interrupt */
        if (efx_nic_rev(efx) >= EFX_REV_FALCON_B0)