Setting type when gpu sub node is discovered

Este cometimento está contido em:
Jack Snyder criou e Sylvain Jeaugey cometeu 2020-08-05 13:39:23 -07:00
1 ascendente 3d63f89068
cometimento de49a77074
1 ficheiro modificado
+1
+1
Ver ficheiro
@@ -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;