ice_vsi_map_rings_to_vectors(vsi);
 
-               /* Associate q_vector rings to napi */
-               ice_vsi_set_napi_queues(vsi);
-
                vsi->stat_offsets_loaded = false;
 
                /* ICE_VSI_CTRL does not need RSS so skip RSS processing */
        if (!test_and_set_bit(ICE_VSI_DOWN, vsi->state))
                ice_down(vsi);
 
+       ice_vsi_clear_napi_queues(vsi);
        ice_vsi_free_irq(vsi);
        ice_vsi_free_tx_rings(vsi);
        ice_vsi_free_rx_rings(vsi);
 }
 
 /**
- * __ice_queue_set_napi - Set the napi instance for the queue
- * @dev: device to which NAPI and queue belong
- * @queue_index: Index of queue
- * @type: queue type as RX or TX
- * @napi: NAPI context
- * @locked: is the rtnl_lock already held
- *
- * Set the napi instance for the queue. Caller indicates the lock status.
- */
-static void
-__ice_queue_set_napi(struct net_device *dev, unsigned int queue_index,
-                    enum netdev_queue_type type, struct napi_struct *napi,
-                    bool locked)
-{
-       if (!locked)
-               rtnl_lock();
-       netif_queue_set_napi(dev, queue_index, type, napi);
-       if (!locked)
-               rtnl_unlock();
-}
-
-/**
- * ice_queue_set_napi - Set the napi instance for the queue
- * @vsi: VSI being configured
- * @queue_index: Index of queue
- * @type: queue type as RX or TX
- * @napi: NAPI context
+ * ice_vsi_set_napi_queues - associate netdev queues with napi
+ * @vsi: VSI pointer
  *
- * Set the napi instance for the queue. The rtnl lock state is derived from the
- * execution path.
+ * Associate queue[s] with napi for all vectors.
+ * The caller must hold rtnl_lock.
  */
-void
-ice_queue_set_napi(struct ice_vsi *vsi, unsigned int queue_index,
-                  enum netdev_queue_type type, struct napi_struct *napi)
+void ice_vsi_set_napi_queues(struct ice_vsi *vsi)
 {
-       struct ice_pf *pf = vsi->back;
+       struct net_device *netdev = vsi->netdev;
+       int q_idx, v_idx;
 
-       if (!vsi->netdev)
+       if (!netdev)
                return;
 
-       if (current_work() == &pf->serv_task ||
-           test_bit(ICE_PREPARED_FOR_RESET, pf->state) ||
-           test_bit(ICE_DOWN, pf->state) ||
-           test_bit(ICE_SUSPENDED, pf->state))
-               __ice_queue_set_napi(vsi->netdev, queue_index, type, napi,
-                                    false);
-       else
-               __ice_queue_set_napi(vsi->netdev, queue_index, type, napi,
-                                    true);
-}
+       ice_for_each_rxq(vsi, q_idx)
+               netif_queue_set_napi(netdev, q_idx, NETDEV_QUEUE_TYPE_RX,
+                                    &vsi->rx_rings[q_idx]->q_vector->napi);
 
-/**
- * __ice_q_vector_set_napi_queues - Map queue[s] associated with the napi
- * @q_vector: q_vector pointer
- * @locked: is the rtnl_lock already held
- *
- * Associate the q_vector napi with all the queue[s] on the vector.
- * Caller indicates the lock status.
- */
-void __ice_q_vector_set_napi_queues(struct ice_q_vector *q_vector, bool locked)
-{
-       struct ice_rx_ring *rx_ring;
-       struct ice_tx_ring *tx_ring;
-
-       ice_for_each_rx_ring(rx_ring, q_vector->rx)
-               __ice_queue_set_napi(q_vector->vsi->netdev, rx_ring->q_index,
-                                    NETDEV_QUEUE_TYPE_RX, &q_vector->napi,
-                                    locked);
-
-       ice_for_each_tx_ring(tx_ring, q_vector->tx)
-               __ice_queue_set_napi(q_vector->vsi->netdev, tx_ring->q_index,
-                                    NETDEV_QUEUE_TYPE_TX, &q_vector->napi,
-                                    locked);
+       ice_for_each_txq(vsi, q_idx)
+               netif_queue_set_napi(netdev, q_idx, NETDEV_QUEUE_TYPE_TX,
+                                    &vsi->tx_rings[q_idx]->q_vector->napi);
        /* Also set the interrupt number for the NAPI */
-       netif_napi_set_irq(&q_vector->napi, q_vector->irq.virq);
-}
+       ice_for_each_q_vector(vsi, v_idx) {
+               struct ice_q_vector *q_vector = vsi->q_vectors[v_idx];
 
-/**
- * ice_q_vector_set_napi_queues - Map queue[s] associated with the napi
- * @q_vector: q_vector pointer
- *
- * Associate the q_vector napi with all the queue[s] on the vector
- */
-void ice_q_vector_set_napi_queues(struct ice_q_vector *q_vector)
-{
-       struct ice_rx_ring *rx_ring;
-       struct ice_tx_ring *tx_ring;
-
-       ice_for_each_rx_ring(rx_ring, q_vector->rx)
-               ice_queue_set_napi(q_vector->vsi, rx_ring->q_index,
-                                  NETDEV_QUEUE_TYPE_RX, &q_vector->napi);
-
-       ice_for_each_tx_ring(tx_ring, q_vector->tx)
-               ice_queue_set_napi(q_vector->vsi, tx_ring->q_index,
-                                  NETDEV_QUEUE_TYPE_TX, &q_vector->napi);
-       /* Also set the interrupt number for the NAPI */
-       netif_napi_set_irq(&q_vector->napi, q_vector->irq.virq);
+               netif_napi_set_irq(&q_vector->napi, q_vector->irq.virq);
+       }
 }
 
 /**
- * ice_vsi_set_napi_queues
+ * ice_vsi_clear_napi_queues - dissociate netdev queues from napi
  * @vsi: VSI pointer
  *
- * Associate queue[s] with napi for all vectors
+ * Clear the association between all VSI queues queue[s] and napi.
+ * The caller must hold rtnl_lock.
  */
-void ice_vsi_set_napi_queues(struct ice_vsi *vsi)
+void ice_vsi_clear_napi_queues(struct ice_vsi *vsi)
 {
-       int i;
+       struct net_device *netdev = vsi->netdev;
+       int q_idx;
 
-       if (!vsi->netdev)
+       if (!netdev)
                return;
 
-       ice_for_each_q_vector(vsi, i)
-               ice_q_vector_set_napi_queues(vsi->q_vectors[i]);
+       ice_for_each_txq(vsi, q_idx)
+               netif_queue_set_napi(netdev, q_idx, NETDEV_QUEUE_TYPE_TX, NULL);
+
+       ice_for_each_rxq(vsi, q_idx)
+               netif_queue_set_napi(netdev, q_idx, NETDEV_QUEUE_TYPE_RX, NULL);
 }
 
 /**