1 files changed, 9 insertions, 0 deletions

diff --git a/drivers/pci/pcie/pme.c b/drivers/pci/pcie/pme.c
index 0057344a3fcb..001f1b78f39c 100644
--- a/drivers/pci/pcie/pme.c
+++ b/drivers/pci/pcie/pme.c
@@ -84,6 +84,9 @@ static bool pcie_pme_walk_bus(struct pci_bus *bus)
         list_for_each_entry(dev, &bus->devices, bus_list) {
                 /* Skip PCIe devices in case we started from a root port. */
                 if (!pci_is_pcie(dev) && pci_check_pme_status(dev)) {
+                        if (dev->pme_poll)
+                                dev->pme_poll = false;
+
                         pci_wakeup_event(dev);
                         pm_request_resume(&dev->dev);
                         ret = true;
@@ -142,6 +145,9 @@ static void pcie_pme_handle_request(struct pci_dev *port, u16 req_id)
 
         /* First, check if the PME is from the root port itself. */
         if (port->devfn == devfn && port->bus->number == busnr) {
+                if (port->pme_poll)
+                        port->pme_poll = false;
+
                 if (pci_check_pme_status(port)) {
                         pm_request_resume(&port->dev);
                         found = true;
@@ -187,6 +193,9 @@ static void pcie_pme_handle_request(struct pci_dev *port, u16 req_id)
                 /* The device is there, but we have to check its PME status. */
                 found = pci_check_pme_status(dev);
                 if (found) {
+                        if (dev->pme_poll)
+                                dev->pme_poll = false;
+
                         pci_wakeup_event(dev);
                         pm_request_resume(&dev->dev);
                 }