Adjust things so we can load cuda libraries on macOS.

Author: abergeron

src/loaders/libcublas.c

@@ -41,6 +41,7 @@ int load_libcublas(int major, int minor) {
 
 #if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
   {
+    static const char DIGITS[] = "0123456789";
     char libname[] = "cublas64_??.dll";
 
     libname[9] = DIGITS[major];
@@ -49,7 +50,17 @@ int load_libcublas(int major, int minor) {
     lib = ga_load_library(libname);
   }
 #else /* Unix */
+#ifdef __APPLE__
+  {
+    static const char DIGITS[] = "0123456789";
+    char libname[] = "/Developer/NVIDIA/CUDA-?.?/lib/libcublas.dylib";
+    libname[23] = DIGITS[major];
+    libname[25] = DIGITS[minor];
+    lib = ga_load_library(libname);
+  }
+#else
   lib = ga_load_library("libcublas.so");
+#endif
 #endif
   if (lib == NULL)
     return GA_LOAD_ERROR;

src/loaders/libcuda.c

@@ -8,8 +8,12 @@
 #if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
 static char libname[] = "nvcuda.dll";
 #else /* Unix */
+#ifdef __APPLE__
+static char libname[] = "CUDA.framework/CUDA";
+#else
 static char libname[] = "libcuda.so";
 #endif
+#endif
 
 #define DEF_PROC(name, args) t##name *name
 #define DEF_PROC_V2(name, args) DEF_PROC(name, args)

src/loaders/libnccl.c

@@ -4,18 +4,21 @@
 #include "dyn_load.h"
 #include "gpuarray/error.h"
 
-#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
-static const char libname[] = "nccl.dll";
-#else /* Unix */
-static const char libname[] = "libnccl.so";
-#endif
-
 #define DEF_PROC(ret, name, args) t##name *name
 
 #include "libnccl.fn"
 
 #undef DEF_PROC
 
+#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64) || defined(__APPLE__)
+/* As far as we know, nccl is not available or buildable on platforms
+   other than linux */
+int load_libnccl(void) {
+  return GA_UNSUPPORTED_ERROR;
+}
+#else /* Unix */
+static const char libname[] = "libnccl.so";
+
 #define DEF_PROC(ret, name, args)            \
   name = (t##name *)ga_func_ptr(lib, #name); \
   if (name == NULL) {                        \
@@ -39,3 +42,4 @@ int load_libnccl(void) {
   loaded = 1;
   return GA_NO_ERROR;
 }
+#endif

src/loaders/libnvrtc.c

@@ -27,6 +27,7 @@ int load_libnvrtc(int major, int minor) {
 
 #if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
   {
+    static const char DIGITS[] = "0123456789";
     char libname[] = "nvrtc64_??.dll";
 
     libname[8] = DIGITS[major];
@@ -35,7 +36,18 @@ int load_libnvrtc(int major, int minor) {
     lib = ga_load_library(libname);
   }
 #else /* Unix */
+#ifdef __APPLE__
+  {
+    static const char DIGITS[] = "0123456789";
+    /* Try the usual fullpath first */
+    char libname[] = "/Developer/NVIDIA/CUDA-?.?/lib/libnvrtc.dylib";
+    libname[23] = DIGITS[major];
+    libname[25] = DIGITS[minor];
+    lib = ga_load_library(libname);
+  }
+#else
   lib = ga_load_library("libnvrtc.so");
+#endif
 #endif
   if (lib == NULL)
     return GA_LOAD_ERROR;