1 files changed, 22 insertions, 0 deletions

diff --git a/arch/powerpc/platforms/pseries/pci.c b/arch/powerpc/platforms/pseries/pci.c
index 41d8a4d1d02e..7725825d887d 100644
--- a/arch/powerpc/platforms/pseries/pci.c
+++ b/arch/powerpc/platforms/pseries/pci.c
@@ -29,6 +29,7 @@
 #include <asm/pci-bridge.h>
 #include <asm/prom.h>
 #include <asm/ppc-pci.h>
+#include <asm/pci.h>
 #include "pseries.h"
 
 #if 0
@@ -237,6 +238,8 @@ static void __init pSeries_request_regions(void)
 
 void __init pSeries_final_fixup(void)
 {
+        struct pci_controller *hose;
+
         pSeries_request_regions();
 
         eeh_probe_devices();
@@ -246,6 +249,25 @@ void __init pSeries_final_fixup(void)
         ppc_md.pcibios_sriov_enable = pseries_pcibios_sriov_enable;
         ppc_md.pcibios_sriov_disable = pseries_pcibios_sriov_disable;
 #endif
+        list_for_each_entry(hose, &hose_list, list_node) {
+                struct device_node *dn = hose->dn, *nvdn;
+
+                while (1) {
+                        dn = of_find_all_nodes(dn);
+                        if (!dn)
+                                break;
+                        nvdn = of_parse_phandle(dn, "ibm,nvlink", 0);
+                        if (!nvdn)
+                                continue;
+                        if (!of_device_is_compatible(nvdn, "ibm,npu-link"))
+                                continue;
+                        if (!of_device_is_compatible(nvdn->parent,
+                                                "ibm,power9-npu"))
+                                continue;
+                        WARN_ON_ONCE(pnv_npu2_init(hose));
+                        break;
+                }
+        }
 }
 
 /*