yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Libwrap checks for LIB.so.1 if LIB.so not found

Browse Source 
Change-Id: I6f07f887f828cb2259dcfd496a2ad707db898cf5
Reviewed-on: http://git-master/r/1000162
Reviewed-by: Przemek Tredak <ptredak@nvidia.com>
Tested-by: Przemek Tredak <ptredak@nvidia.com>
This commit is contained in:
Nathan Luehr 2016-01-21 16:45:19 -08:00 committed by Przemek Tredak
parent 2758353380
commit caa40b8dd3
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/libwrap.cu
View File

@ -59,14 +59,20 @@ ncclResult_t wrapSymbols(void) {
nvmlhandle=dlopen("libnvidia-ml.so", RTLD_NOW); nvmlhandle=dlopen("libnvidia-ml.so", RTLD_NOW);
if (!nvmlhandle) { if (!nvmlhandle) {
WARN("Failed to open libnvidia-ml.so"); nvmlhandle=dlopen("libnvidia-ml.so.1", RTLD_NOW);
goto teardown; if (!nvmlhandle) {
WARN("Failed to open libnvidia-ml.so[.1]");
goto teardown;
}
} }
cuhandle = dlopen("libcuda.so", RTLD_NOW); cuhandle = dlopen("libcuda.so", RTLD_NOW);
if (!cuhandle) { if (!cuhandle) {
WARN("Failed to open libcuda.so"); cuhandle = dlopen("libcuda.so.1", RTLD_NOW);
goto teardown; if (!cuhandle) {
WARN("Failed to open libcuda.so[.1]");
goto teardown;
}
} }
#define LOAD_SYM(handle, symbol, funcptr) do { \ #define LOAD_SYM(handle, symbol, funcptr) do { \