AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change network plugin name to librccl-net.so (#266)

Browse Source 
[ROCm/rccl commit: d871fceb54]
This commit is contained in:
Wenkai Du
2020-09-18 13:23:30 -07:00
committed by GitHub
parent 2e5e4a6bde
commit 37f7eec6b7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/init.cc
+2 -2
View File
@@ -70,13 +70,13 @@ ncclResult_t initCollNet(ncclCollNet_t* collnet) {
} }
ncclResult_t initNetPlugin(ncclNet_t** net, ncclCollNet_t** collnet) { ncclResult_t initNetPlugin(ncclNet_t** net, ncclCollNet_t** collnet) {
void* netPluginLib = dlopen("libnccl-net.so", RTLD_NOW | RTLD_LOCAL); void* netPluginLib = dlopen("librccl-net.so", RTLD_NOW | RTLD_LOCAL);
if (netPluginLib == NULL) { if (netPluginLib == NULL) {
// dlopen does not guarantee to set errno, but dlerror only gives us a // dlopen does not guarantee to set errno, but dlerror only gives us a
// string, so checking errno doesn't hurt to try to provide a better // string, so checking errno doesn't hurt to try to provide a better
// error message // error message
if (errno == ENOENT) { if (errno == ENOENT) {
INFO(NCCL_INIT|NCCL_NET, "NET/Plugin : No plugin found (libnccl-net.so), using internal implementation"); INFO(NCCL_INIT|NCCL_NET, "NET/Plugin : No plugin found (librccl-net.so), using internal implementation");
} else { } else {
INFO(NCCL_INIT|NCCL_NET, "NET/Plugin : Plugin load returned %d : %s.", errno, dlerror()); INFO(NCCL_INIT|NCCL_NET, "NET/Plugin : Plugin load returned %d : %s.", errno, dlerror());
} }