Setting type when gpu sub node is discovered

[ROCm/rccl commit: de49a77074]
This commit is contained in:
Jack Snyder
2020-08-05 13:34:32 -07:00
committed by Sylvain Jeaugey
orang tua 149bf427d5
melakukan dca19952fd
+1
Melihat File
@@ -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;