Minor fix for some msccl installations (#862)

[ROCm/rccl commit: 5bcd3768cc]
This commit is contained in:
gilbertlee-amd
2023-08-23 13:48:58 -06:00
committed by GitHub
parent 5e60fb93d5
commit 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();
}