x86: Add CPU detection for Zhaoxin processors

Signed-off-by: Jonas Zhou <JonasZhou@zhaoxin.com>
2020-11-27 16:20:55 +08:00
parent 920dbe5b35
commit 3996562690
4 changed files with 16 additions and 1 deletions
@@ -283,6 +283,9 @@ ncclResult_t ncclTopoCheckP2p(struct ncclTopoSystem* system, int64_t id1, int64_
    if (model == NCCL_TOPO_CPU_TYPE_BDW) p2pLevel = PATH_PXB;
    else p2pLevel = PATH_PHB;
  }
+  if (arch == NCCL_TOPO_CPU_ARCH_X86 && vendor == NCCL_TOPO_CPU_VENDOR_ZHAOXIN) {
+    p2pLevel = PATH_PXB;
+  }

 compare:
  // Compute the PCI distance and compare with the p2pLevel.