Fix #951: Fallback gracefully if a library can't be loaded

Author: mdboom

cuda_bindings/cuda/bindings/_internal/cufile_linux.pyx

@@ -9,7 +9,7 @@ import threading
 
 from .utils import FunctionNotFoundError, NotSupportedError
 
-from cuda.pathfinder import load_nvidia_dynamic_lib
+from cuda.pathfinder import load_nvidia_dynamic_lib, DynamicLibNotFoundError
 
 import cython
 
@@ -104,7 +104,11 @@ cdef void* __cuFileGetParameterPosixPoolSlabArray = NULL
 
 
 cdef void* load_library() except* with gil:
-    cdef uintptr_t handle = load_nvidia_dynamic_lib("cufile")._handle_uint
+    cdef uintptr_t handle
+    try:
+        handle = load_nvidia_dynamic_lib("cufile")._handle_uint
+    except DynamicLibNotFoundError:
+        handle = 0
     return <void*>handle
 
 

cuda_bindings/cuda/bindings/_internal/nvjitlink_linux.pyx

@@ -9,7 +9,7 @@ from libc.stdint cimport intptr_t, uintptr_t
 import threading
 from .utils import FunctionNotFoundError, NotSupportedError
 
-from cuda.pathfinder import load_nvidia_dynamic_lib
+from cuda.pathfinder import load_nvidia_dynamic_lib, DynamicLibNotFoundError
 
 
 ###############################################################################
@@ -73,7 +73,11 @@ cdef void* __nvJitLinkVersion = NULL
 
 
 cdef void* load_library() except* with gil:
-    cdef uintptr_t handle = load_nvidia_dynamic_lib("nvJitLink")._handle_uint
+    cdef uintptr_t handle
+    try:
+        handle = load_nvidia_dynamic_lib("nvJitLink")._handle_uint
+    except DynamicLibNotFoundError:
+        handle = 0
     return <void*>handle
 
 

cuda_bindings/cuda/bindings/_internal/nvjitlink_windows.pyx

@@ -4,12 +4,12 @@
 #
 # This code was automatically generated across versions from 12.0.1 to 13.0.1. Do not modify it directly.
 
-from libc.stdint cimport intptr_t
+from libc.stdint cimport intptr_t, uintptr_t
 
 import threading
 from .utils import FunctionNotFoundError, NotSupportedError
 
-from cuda.pathfinder import load_nvidia_dynamic_lib
+from cuda.pathfinder import load_nvidia_dynamic_lib, DynamicLibNotFoundError
 
 from libc.stddef cimport wchar_t
 from libc.stdint cimport uintptr_t
@@ -95,12 +95,14 @@ cdef void* __nvJitLinkVersion = NULL
 
 cdef int __check_or_init_nvjitlink() except -1 nogil:
     global __py_nvjitlink_init
-    if __py_nvjitlink_init:
-        return 0
+    cdef uintptr_t handle
 
     with gil, __symbol_lock:
         # Load library
-        handle = load_nvidia_dynamic_lib("nvJitLink")._handle_uint
+        try:
+            handle = load_nvidia_dynamic_lib("nvJitLink")._handle_uint
+        except DynamicLibNotFoundError:
+            handle = 0
 
         # Load function
         global __nvJitLinkCreate

cuda_bindings/cuda/bindings/_internal/nvvm_linux.pyx

@@ -9,7 +9,7 @@ from libc.stdint cimport intptr_t, uintptr_t
 import threading
 from .utils import FunctionNotFoundError, NotSupportedError
 
-from cuda.pathfinder import load_nvidia_dynamic_lib
+from cuda.pathfinder import load_nvidia_dynamic_lib, DynamicLibNotFoundError
 
 
 ###############################################################################
@@ -72,7 +72,11 @@ cdef void* __nvvmGetProgramLog = NULL
 
 
 cdef void* load_library() except* with gil:
-    cdef uintptr_t handle = load_nvidia_dynamic_lib("nvvm")._handle_uint
+    cdef uintptr_t handle
+    try:
+        handle = load_nvidia_dynamic_lib("nvvm")._handle_uint
+    except DynamicLibNotFoundError:
+        handle = 0
     return <void*>handle
 
 
@@ -178,7 +182,7 @@ cdef int __check_or_init_nvvm() except -1 nogil:
         return 0
 
 
-cdef inline int _check_or_init_nvvm() except -1 nogil:
+cdef int _check_or_init_nvvm() except -1 nogil:
     if __py_nvvm_init:
         return 0
 

cuda_bindings/cuda/bindings/_internal/nvvm_windows.pyx

@@ -4,12 +4,12 @@
 #
 # This code was automatically generated across versions from 12.0.1 to 13.0.1. Do not modify it directly.
 
-from libc.stdint cimport intptr_t
+from libc.stdint cimport intptr_t, uintptr_t
 
 import threading
 from .utils import FunctionNotFoundError, NotSupportedError
 
-from cuda.pathfinder import load_nvidia_dynamic_lib
+from cuda.pathfinder import load_nvidia_dynamic_lib, DynamicLibNotFoundError
 
 from libc.stddef cimport wchar_t
 from libc.stdint cimport uintptr_t
@@ -95,9 +95,14 @@ cdef void* __nvvmGetProgramLog = NULL
 cdef int __check_or_init_nvvm() except -1 nogil:
     global __py_nvvm_init
 
+    cdef uintptr_t handle
+
     with gil, __symbol_lock:
         # Load library
-        handle = load_nvidia_dynamic_lib("nvvm")._handle_uint
+        try:
+            handle = load_nvidia_dynamic_lib("nvvm")._handle_uint
+        except DynamicLibNotFoundError:
+            handle = 0
 
         # Load function
         global __nvvmGetErrorString