PCI: portdrv: move PME capability check

No reason to check PME capability outside get_port_device_capability().
Do it in get_port_device_capability().

Signed-off-by: Kenji Kaneshige <kaneshige.kenji@jp.fujitsu.com>
Signed-off-by: Jesse Barnes <jbarnes@virtuousgeek.org>

1 files changed, 3 insertions, 3 deletions

diff --git a/drivers/pci/pcie/portdrv_core.c b/drivers/pci/pcie/portdrv_core.c
index 42b21eec15f0..079bbc3ed4f8 100644
--- a/drivers/pci/pcie/portdrv_core.c
+++ b/drivers/pci/pcie/portdrv_core.c
@@ -241,6 +241,9 @@ static int get_port_device_capability(struct pci_dev *dev)
         /* VC support */
         if (pci_find_ext_capability(dev, PCI_EXT_CAP_ID_VC))
                 services |= PCIE_PORT_SERVICE_VC;
+        /* Root ports are capable of generating PME too */
+        if (dev->pcie_type == PCI_EXP_TYPE_ROOT_PORT)
+                services |= PCIE_PORT_SERVICE_PME;
 
         return services;
 }
@@ -302,9 +305,6 @@ int pcie_port_device_register(struct pci_dev *dev)
         port_data->port_type = dev->pcie_type;
 
         capabilities = get_port_device_capability(dev);
-        /* Root ports are capable of generating PME too */
-        if (port_data->port_type == PCIE_RC_PORT)
-                capabilities |= PCIE_PORT_SERVICE_PME;
 
         irq_mode = assign_interrupt_mode(dev, vectors, capabilities);
         if (irq_mode == PCIE_PORT_NO_IRQ) {