We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 746cee0 commit 08c3041Copy full SHA for 08c3041
1 file changed
cuda_bindings/cuda/bindings/_internal/nvjitlink_linux.pyx
@@ -59,7 +59,10 @@ cdef void* load_library(const int driver_ver) except* with gil:
59
paths = path_finder.get_cuda_paths()
60
paths_cudalib_dir = paths["cudalib_dir"]
61
if paths_cudalib_dir:
62
+ # TODO(rwgk): Produce the correct so_name in path_finder.py
63
so_name = os.path.join(paths_cudalib_dir.info, "libnvJitLink.so")
64
+ if not os.path.exists(so_name) and so_name.count("/lib64/") == 1:
65
+ so_name = so_name.replace("/lib64/", "/lib/")
66
handle = dlopen(so_name.encode(), RTLD_NOW | RTLD_GLOBAL)
67
if handle == NULL:
68
err_msg = dlerror()
0 commit comments