Fix crash when only a subset of GPUs are visible within a container.

Fixes #326.


[ROCm/rccl commit: f36540f55a]
Этот коммит содержится в:
Sylvain Jeaugey
2020-04-17 10:03:14 -07:00
родитель 5df2502deb
Коммит c43022b9d8
2 изменённых файлов: 2 добавлений и 1 удалений
+1 -1
Просмотреть файл
@@ -569,7 +569,7 @@ ncclResult_t ncclTopoGetXmlFromGpu(struct ncclXmlNode* pciNode, nvmlDevice_t nvm
NCCLCHECK(xmlSetAttrInt(gpuNode, "dev", dev));
}
NCCLCHECK(xmlGetAttrInt(gpuNode, "dev", &dev));
if (dev == -1) return ncclSuccess;
if (dev == -1) { *gpuNodeRet = NULL; return ncclSuccess; }
NCCLCHECK(xmlGetAttrIndex(gpuNode, "sm", &index));
if (index == -1) {