Fix path finding in msccl internal scheduler (#794)

This commit is contained in:
Ziyue Yang authored and GitHub committed 2023-07-12 13:31:28 -07:00
1 parent 43f13cd25a
commit 848e60b00c
1 file changed
+1 -1
+1 -1
View File
@@ -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();
}