yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/master'

Browse Source
This commit is contained in:
Sylvain Jeaugey 2023-11-20 05:07:23 -08:00
commit b6d7438d31
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/graph/xml.cc
View File

@ -13,6 +13,9 @@
#include "core.h" #include "core.h"
#include "nvmlwrap.h" #include "nvmlwrap.h"
#include "xml.h" #include "xml.h"
#if defined(__x86_64__)
#include <cpuid.h>
#endif
/*******************/ /*******************/
/* XML File Parser */ /* XML File Parser */
@ -413,7 +416,8 @@ ncclResult_t ncclTopoGetXmlFromCpu(struct ncclXmlNode* cpuNode, struct ncclXml*
char vendor[12]; char vendor[12];
} cpuid0; } cpuid0;
asm volatile("cpuid" : "=b" (cpuid0.ebx), "=c" (cpuid0.ecx), "=d" (cpuid0.edx) : "a" (0) : "memory"); unsigned unused;
__cpuid(0, unused, cpuid0.ebx, cpuid0.ecx, cpuid0.edx);
char vendor[13]; char vendor[13];
strncpy(vendor, cpuid0.vendor, 12); strncpy(vendor, cpuid0.vendor, 12);
vendor[12] = '\0'; vendor[12] = '\0';
@ -435,7 +439,8 @@ ncclResult_t ncclTopoGetXmlFromCpu(struct ncclXmlNode* cpuNode, struct ncclXml*
}; };
uint32_t val; uint32_t val;
} cpuid1; } cpuid1;
asm volatile("cpuid" : "=a" (cpuid1.val) : "a" (1) : "memory"); unsigned unused;
__cpuid(1, cpuid1.val, unused, unused, unused);
int familyId = cpuid1.familyId + (cpuid1.extFamilyId << 4); int familyId = cpuid1.familyId + (cpuid1.extFamilyId << 4);
int modelId = cpuid1.modelId + (cpuid1.extModelId << 4); int modelId = cpuid1.modelId + (cpuid1.extModelId << 4);
NCCLCHECK(xmlSetAttrInt(cpuNode, "familyid", familyId)); NCCLCHECK(xmlSetAttrInt(cpuNode, "familyid", familyId));