Fix path finding in msccl internal scheduler (#794)

[ROCm/rccl commit: 848e60b00c]
This commit is contained in:
Ziyue Yang
2023-07-13 04:31:28 +08:00
committed by GitHub
parent a5feda50b9
commit 17926777ff
@@ -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();
}