Minor fix for some msccl installations (#862)
[ROCm/rccl commit: 5bcd3768cc]
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
5e60fb93d5
Коммит
3dd880fe74
@@ -98,7 +98,7 @@ static ncclResult_t mscclInternalSchedulerInit() {
|
||||
mscclAlgoDirStr += (mscclUnitTestMode && mscclUnitTestMode()) ? mscclUnitTestAlgoDefaultDir : mscclAlgoDefaultDir;
|
||||
mscclAlgoDir = mscclAlgoDirStr.c_str();
|
||||
// Get share Directory Paths
|
||||
mscclAlgoShareDirStr = selfLibPath.substr(0, selfLibPath.find("lib"));
|
||||
mscclAlgoShareDirStr = selfLibPath.substr(0, selfLibPath.rfind("lib"));
|
||||
mscclAlgoShareDirStr += (mscclUnitTestMode && mscclUnitTestMode()) ? mscclUnitTestAlgoShareDirPath : mscclAlgoShareDirPath;
|
||||
mscclAlgoShareDir = mscclAlgoShareDirStr.c_str();
|
||||
}
|
||||
|
||||
Ссылка в новой задаче
Block a user