Setting type when gpu sub node is discovered

[ROCm/rccl commit: de49a77074]
Tento commit je obsažen v:
Jack Snyder
2020-08-05 13:34:32 -07:00
odevzdal Sylvain Jeaugey
rodič 149bf427d5
revize dca19952fd
+1
Zobrazit soubor
@@ -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;