We already changed NPU API for GPUs to not to call OPAL and the remaining
bit is initializing NPU structures.

This searches for POWER9 NVLinks attached to any device on a PHB and
initializes an NPU structure if any found.

Signed-off-by: Alexey Kardashevskiy <a...@ozlabs.ru>
---
Changes:
v5:
* added WARN_ON_ONCE

v4:
* dropped "IBM,npu-vphb" compatible type on PHB and use the type of NVLink
---
 arch/powerpc/platforms/pseries/pci.c | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/arch/powerpc/platforms/pseries/pci.c 
b/arch/powerpc/platforms/pseries/pci.c
index 41d8a4d..7725825 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;
+               }
+       }
 }
 
 /*
-- 
2.17.1

Reply via email to