AI/rocm-systems
2
0 Garpu 0
Kode Masalah Tarik Permintaan Actions Packages Projects Rilis Wiki Kegiatan

x86: Add CPU detection for Zhaoxin processors

Telusuri Sumber 
Signed-off-by: Jonas Zhou <JonasZhou@zhaoxin.com>
This commit is contained in:
Jonas Zhou authored and Sylvain Jeaugey committed 2020-12-17 11:15:18 -08:00
1 orang tua 920dbe5b35
melakukan 3996562690
4 files changed
+16 -1
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/include/graph.h
+2
Melihat File
@@ -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);