msi_lock_descs(data->dev);
        irq_domain_free_irqs_common(domain, virq, nr_irqs);
-       msi_free_msi_descs_range(data->dev, MSI_DESC_ALL, virq, virq + nr_irqs - 1);
+       msi_free_msi_descs_range(data->dev, virq, virq + nr_irqs - 1);
        msi_unlock_descs(data->dev);
 }
 
 
 #endif /* CONFIG_PCI_MSI */
 
 int msi_add_msi_desc(struct device *dev, struct msi_desc *init_desc);
-void msi_free_msi_descs_range(struct device *dev, enum msi_desc_filter filter,
-                             unsigned int first_index, unsigned int last_index);
+void msi_free_msi_descs_range(struct device *dev, unsigned int first_index, unsigned int last_index);
 
 /**
  * msi_free_msi_descs - Free MSI descriptors of a device
  */
 static inline void msi_free_msi_descs(struct device *dev)
 {
-       msi_free_msi_descs_range(dev, MSI_DESC_ALL, 0, MSI_MAX_INDEX);
+       msi_free_msi_descs_range(dev, 0, MSI_MAX_INDEX);
 }
 
 void __pci_read_msi_msg(struct msi_desc *entry, struct msi_msg *msg);
 
 fail_mem:
        ret = -ENOMEM;
 fail:
-       msi_free_msi_descs_range(dev, MSI_DESC_ALL, index, last);
+       msi_free_msi_descs_range(dev, index, last);
        return ret;
 }
 
 /**
  * msi_free_msi_descs_range - Free MSI descriptors of a device
  * @dev:               Device to free the descriptors
- * @filter:            Descriptor state filter
  * @first_index:       Index to start freeing from
  * @last_index:                Last index to be freed
  */
-void msi_free_msi_descs_range(struct device *dev, enum msi_desc_filter filter,
-                             unsigned int first_index, unsigned int last_index)
+void msi_free_msi_descs_range(struct device *dev, unsigned int first_index,
+                             unsigned int last_index)
 {
        struct xarray *xa = &dev->msi.data->__store;
        struct msi_desc *desc;
        lockdep_assert_held(&dev->msi.data->mutex);
 
        xa_for_each_range(xa, idx, desc, first_index, last_index) {
-               if (msi_desc_match(desc, filter)) {
-                       xa_erase(xa, idx);
-                       msi_free_desc(desc);
-               }
+               xa_erase(xa, idx);
+
+               /* Leak the descriptor when it is still referenced */
+               if (WARN_ON_ONCE(msi_desc_match(desc, MSI_DESC_ASSOCIATED)))
+                       continue;
+               msi_free_desc(desc);
        }
 }
 
 fail:
        for (--virq; virq >= virq_base; virq--)
                irq_domain_free_irqs_common(domain, virq, 1);
-       msi_free_msi_descs_range(dev, MSI_DESC_ALL, virq_base, virq_base + nvec - 1);
+       msi_free_msi_descs_range(dev, virq_base, virq_base + nvec - 1);
 unlock:
        msi_unlock_descs(dev);
        return ret;