diff options
author | Jack Snyder <jsnyder@prom.nvidia.com> | 2020-08-05 23:34:32 +0300 |
---|---|---|
committer | Sylvain Jeaugey <sjeaugey@nvidia.com> | 2020-08-05 23:39:23 +0300 |
commit | de49a77074e884758e689ff0204f066eac7aae46 (patch) | |
tree | d69147164b21cc279002e45b834de732d81afcdc | |
parent | 3d63f8906882418c33ba4d1fbd4567ba29acbeba (diff) |
Setting type when gpu sub node is discovered
-rw-r--r-- | src/graph/topo.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/graph/topo.cc b/src/graph/topo.cc index a917340..ed4bd23 100644 --- a/src/graph/topo.cc +++ b/src/graph/topo.cc @@ -320,6 +320,7 @@ ncclResult_t ncclTopoAddPci(struct ncclXmlNode* xmlPci, struct ncclTopoSystem* s struct ncclXmlNode* xmlGpu = NULL; NCCLCHECK(xmlGetSub(xmlPci, "gpu", &xmlGpu)); if (xmlGpu != NULL) { + type = GPU; int index; NCCLCHECK(xmlGetAttrIndex(xmlGpu, "rank", &index)); if (index == -1) return ncclSuccess; |