Setting type when gpu sub node is discovered

NVIDIA · Aug 5, 2020 · de49a77 · de49a77
1 parent 3d63f89
commit de49a77
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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;