p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix dynamic load error

This commit is contained in:
ZPaC 2021-11-16 11:20:10 +08:00
parent 50bbdfef17
commit 611de83fd8
2 changed files with 1 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
mindspore/ccsrc/runtime/device/gpu/distribution/collective_init.cc
View File

@ -27,12 +27,7 @@ CollectiveInitializer &CollectiveInitializer::instance() {
bool CollectiveInitializer::collective_inited() const { return collective_inited_; }
const void *CollectiveInitializer::collective_handle() {
if (collective_handle_ == nullptr) {
DynamicLoadCollectiveLib();
}
return collective_handle_;
}
const void *CollectiveInitializer::collective_handle() { return collective_handle_; }
void CollectiveInitializer::InitCollective() {
void *handle = dlopen("libgpu_collective.so", RTLD_LAZY);
@ -61,22 +56,6 @@ void CollectiveInitializer::FinalizeCollective() {
}
}
}
void CollectiveInitializer::DynamicLoadCollectiveLib() {
void *handle = dlopen("libgpu_collective.so", RTLD_LAZY);
if (handle == nullptr) {
MS_LOG(EXCEPTION)
<< "Loading libgpu_collective.so failed. Many reasons could cause this:\n"
"1.libgpu_collective.so is not found, please check this MindSpore package is GPU version and built "
"with distributed feature.\n"
"2.NCCL is not found or the user-installed NCCL version installed is incompatible: MindSpore "
"requires NCCL-2.7.6.\n"
"3.OpenMPI is not found or the user-installed OpenMPI version is incompatible: MindSpore "
"requires OpenMPI-4.0.3.\n";
}
collective_inited_ = true;
collective_handle_ = handle;
}
} // namespace gpu
} // namespace device
} // namespace mindspore

3
mindspore/ccsrc/runtime/device/gpu/distribution/collective_init.h
View File

@ -42,9 +42,6 @@ class CollectiveInitializer {
static void InitCollective();
static void FinalizeCollective();
// Dynamically load shared library of libgpu_collective.so.
void DynamicLoadCollectiveLib();
private:
CollectiveInitializer() : collective_inited_(false) {}
~CollectiveInitializer() = default;