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

Setting type when gpu sub node is discovered

Browse Source
This commit is contained in:
Jack Snyder 2020-08-05 13:34:32 -07:00 committed by Sylvain Jeaugey
parent 3d63f89068
commit de49a77074
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/graph/topo.cc
View File

@ -320,6 +320,7 @@ ncclResult_t ncclTopoAddPci(struct ncclXmlNode* xmlPci, struct ncclTopoSystem* s
struct ncclXmlNode* xmlGpu = NULL; struct ncclXmlNode* xmlGpu = NULL;
NCCLCHECK(xmlGetSub(xmlPci, "gpu", &xmlGpu)); NCCLCHECK(xmlGetSub(xmlPci, "gpu", &xmlGpu));
if (xmlGpu != NULL) { if (xmlGpu != NULL) {
type = GPU;
int index; int index;
NCCLCHECK(xmlGetAttrIndex(xmlGpu, "rank", &index)); NCCLCHECK(xmlGetAttrIndex(xmlGpu, "rank", &index));
if (index == -1) return ncclSuccess; if (index == -1) return ncclSuccess;