Fix path finding in msccl internal scheduler (#794)

Этот коммит содержится в:
Ziyue Yang
2023-07-13 04:31:28 +08:00
коммит произвёл GitHub
родитель 43f13cd25a
Коммит 848e60b00c
+1 -1
Просмотреть файл
@@ -93,7 +93,7 @@ static ncclResult_t mscclInternalSchedulerInit() {
mscclAlgoDirStr += (mscclUnitTestMode && mscclUnitTestMode()) ? mscclUnitTestAlgoDefaultDir : mscclAlgoDefaultDir;
mscclAlgoDir = mscclAlgoDirStr.c_str();
// Get share Directory Paths
mscclAlgoShareDirStr = selfLibPath.substr(0, selfLibPath.find_first_of("lib") );
mscclAlgoShareDirStr = selfLibPath.substr(0, selfLibPath.find("lib"));
mscclAlgoShareDirStr += (mscclUnitTestMode && mscclUnitTestMode()) ? mscclUnitTestAlgoShareDirPath : mscclAlgoShareDirPath;
mscclAlgoShareDir = mscclAlgoShareDirStr.c_str();
}