Skip to content

Commit 1239be9

Browse files
committed
Fix logic error in nvjitlink_linux.pyx load_library()
1 parent e421b48 commit 1239be9

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

cuda_bindings/cuda/bindings/_internal/nvjitlink_linux.pyx

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,15 +69,16 @@ cdef void* load_library(const int driver_ver) except* with gil:
6969
raise RuntimeError("Failure obtaining paths_cudalib_dir")
7070
if not paths_cudalib_dir.info:
7171
raise RuntimeError("Failure obtaining paths_cudalib_dir.info")
72-
candidate_so_dirs = [paths_cudalib_dir.info]
72+
primary_so_dir = paths_cudalib_dir.info + "/"
73+
candidate_so_dirs = [primary_so_dir]
7374
libs = ["/lib/", "/lib64/"]
7475
for _ in range(2):
75-
alt_dir = libs[0].join(paths_cudalib_dir.info.rsplit(libs[1], 1))
76+
alt_dir = libs[0].join(primary_so_dir.rsplit(libs[1], 1))
7677
if alt_dir not in candidate_so_dirs:
7778
candidate_so_dirs.append(alt_dir)
7879
libs.reverse()
7980
candidate_so_names = [
80-
os.path.join(so_dirname, so_basename)
81+
so_dirname + so_basename
8182
for so_dirname in candidate_so_dirs]
8283
error_messages = []
8384
for so_name in candidate_so_names:

0 commit comments

Comments
 (0)