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

Change-Id: I6f07f887f828cb2259dcfd496a2ad707db898cf5 Reviewed-on: http://git-master/r/1000162 Reviewed-by: Przemek Tredak <[email protected]> Tested-by: Przemek Tredak <[email protected]>
NVIDIA · Jan 29, 2016 · caa40b8 · caa40b8
1 parent 2758353
commit caa40b8
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/libwrap.cu b/src/libwrap.cu
@@ -59,14 +59,20 @@ ncclResult_t wrapSymbols(void) {
 
   nvmlhandle=dlopen("libnvidia-ml.so", RTLD_NOW);
   if (!nvmlhandle) {
-    WARN("Failed to open libnvidia-ml.so");
-    goto teardown;
+    nvmlhandle=dlopen("libnvidia-ml.so.1", RTLD_NOW);
+    if (!nvmlhandle) {
+      WARN("Failed to open libnvidia-ml.so[.1]");
+      goto teardown;
+    }
   }
 
   cuhandle = dlopen("libcuda.so", RTLD_NOW);
   if (!cuhandle) {
-    WARN("Failed to open libcuda.so");
-    goto teardown;
+    cuhandle = dlopen("libcuda.so.1", RTLD_NOW);
+    if (!cuhandle) {
+      WARN("Failed to open libcuda.so[.1]");
+      goto teardown;
+    }
   }
 
   #define LOAD_SYM(handle, symbol, funcptr) do {         \