Skip to content

Commit 08c3041

Browse files
committed
Change "/lib64/" → "/lib/" in nvjitlink_linux.pyx
1 parent 746cee0 commit 08c3041

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

cuda_bindings/cuda/bindings/_internal/nvjitlink_linux.pyx

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,10 @@ cdef void* load_library(const int driver_ver) except* with gil:
5959
paths = path_finder.get_cuda_paths()
6060
paths_cudalib_dir = paths["cudalib_dir"]
6161
if paths_cudalib_dir:
62+
# TODO(rwgk): Produce the correct so_name in path_finder.py
6263
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/")
6366
handle = dlopen(so_name.encode(), RTLD_NOW | RTLD_GLOBAL)
6467
if handle == NULL:
6568
err_msg = dlerror()

0 commit comments

Comments
 (0)