*/
 int pci_irq_vector(struct pci_dev *dev, unsigned int nr)
 {
-       if (dev->msix_enabled) {
-               struct msi_desc *entry;
-
-               for_each_pci_msi_entry(entry, dev) {
-                       if (entry->msi_index == nr)
-                               return entry->irq;
-               }
-               WARN_ON_ONCE(1);
-               return -EINVAL;
-       }
+       unsigned int irq;
 
-       if (dev->msi_enabled) {
-               struct msi_desc *entry = first_pci_msi_entry(dev);
-
-               if (WARN_ON_ONCE(nr >= entry->nvec_used))
-                       return -EINVAL;
-       } else {
-               if (WARN_ON_ONCE(nr > 0))
-                       return -EINVAL;
-       }
+       if (!dev->msi_enabled && !dev->msix_enabled)
+               return !nr ? dev->irq : -EINVAL;
 
-       return dev->irq + nr;
+       irq = msi_get_virq(&dev->dev, nr);
+       return irq ? irq : -EINVAL;
 }
 EXPORT_SYMBOL(pci_irq_vector);