From caa40b8dd3116fa9d903938b09ea5e047d736307 Mon Sep 17 00:00:00 2001
From: Nathan Luehr <nluehr@nvidia.com>
Date: Thu, 21 Jan 2016 16:45:19 -0800
Subject: [PATCH] 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>
---
 src/libwrap.cu | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

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