1 files changed, 14 insertions, 4 deletions

diff --git a/drivers/pci/pci-driver.c b/drivers/pci/pci-driver.c
index 3f8e3dbcaa7c..d04c5adafc16 100644
--- a/drivers/pci/pci-driver.c
+++ b/drivers/pci/pci-driver.c
@@ -582,7 +582,7 @@ static int pci_legacy_suspend_late(struct device *dev, pm_message_t state)
                         WARN_ONCE(pci_dev->current_state != prev,
                                 "PCI PM: Device state not saved by %pF\n",
                                 drv->suspend_late);
-                        return 0;
+                        goto Fixup;
                 }
         }
 
@@ -591,6 +591,9 @@ static int pci_legacy_suspend_late(struct device *dev, pm_message_t state)
 
         pci_pm_set_unknown_state(pci_dev);
 
+Fixup:
+        pci_fixup_device(pci_fixup_suspend_late, pci_dev);
+
         return 0;
 }
 
@@ -734,7 +737,7 @@ static int pci_pm_suspend_noirq(struct device *dev)
 
         if (!pm) {
                 pci_save_state(pci_dev);
-                return 0;
+                goto Fixup;
         }
 
         if (pm->suspend_noirq) {
@@ -751,7 +754,7 @@ static int pci_pm_suspend_noirq(struct device *dev)
                         WARN_ONCE(pci_dev->current_state != prev,
                                 "PCI PM: State of device not saved by %pF\n",
                                 pm->suspend_noirq);
-                        return 0;
+                        goto Fixup;
                 }
         }
 
@@ -775,6 +778,9 @@ static int pci_pm_suspend_noirq(struct device *dev)
         if (pci_dev->class == PCI_CLASS_SERIAL_USB_EHCI)
                 pci_write_config_word(pci_dev, PCI_COMMAND, 0);
 
+Fixup:
+        pci_fixup_device(pci_fixup_suspend_late, pci_dev);
+
         return 0;
 }
 
@@ -999,8 +1005,10 @@ static int pci_pm_poweroff_noirq(struct device *dev)
         if (pci_has_legacy_pm_support(to_pci_dev(dev)))
                 return pci_legacy_suspend_late(dev, PMSG_HIBERNATE);
 
-        if (!drv || !drv->pm)
+        if (!drv || !drv->pm) {
+                pci_fixup_device(pci_fixup_suspend_late, pci_dev);
                 return 0;
+        }
 
         if (drv->pm->poweroff_noirq) {
                 int error;
@@ -1021,6 +1029,8 @@ static int pci_pm_poweroff_noirq(struct device *dev)
         if (pci_dev->class == PCI_CLASS_SERIAL_USB_EHCI)
                 pci_write_config_word(pci_dev, PCI_COMMAND, 0);
 
+        pci_fixup_device(pci_fixup_suspend_late, pci_dev);
+
         if (pcibios_pm_ops.poweroff_noirq)
                 return pcibios_pm_ops.poweroff_noirq(dev);