2
0

x86: Add CPU detection for Zhaoxin processors

Signed-off-by: Jonas Zhou <JonasZhou@zhaoxin.com>
Este cometimento está contido em:
Jonas Zhou
2020-11-27 16:20:55 +08:00
cometido por Sylvain Jeaugey
ascendente 920dbe5b35
cometimento 3996562690
4 ficheiros modificados com 16 adições e 1 eliminações
+2
Ver ficheiro
@@ -40,8 +40,10 @@ ncclResult_t ncclTopoSetAffinity(struct ncclTopoSystem* system, int rank);
#define NCCL_TOPO_CPU_ARCH_ARM 3
#define NCCL_TOPO_CPU_VENDOR_INTEL 1
#define NCCL_TOPO_CPU_VENDOR_AMD 2
#define NCCL_TOPO_CPU_VENDOR_ZHAOXIN 3
#define NCCL_TOPO_CPU_TYPE_BDW 1
#define NCCL_TOPO_CPU_TYPE_SKL 2
#define NCCL_TOPO_CPU_TYPE_YONGFENG 1
ncclResult_t ncclTopoCpuType(struct ncclTopoSystem* system, int* arch, int* vendor, int* model);
ncclResult_t ncclTopoGetNetCount(struct ncclTopoSystem* system, int* count);