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 <ptredak@nvidia.com>
Tested-by: Przemek Tredak <ptredak@nvidia.com>

1 files changed, 10 insertions, 4 deletions

diff --git a/src/libwrap.cu b/src/libwrap.cu
index 1b3eb75..c4ae737 100644
--- 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 {         \