Setting type when gpu sub node is discovered
[ROCm/rccl commit: de49a77074]
Этот коммит содержится в:
коммит произвёл
Sylvain Jeaugey
родитель
149bf427d5
Коммит
dca19952fd
@@ -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;
|
||||
|
||||
Ссылка в новой задаче
Block a user