File tree Expand file tree Collapse file tree
cuda_bindings/cuda/bindings/_internal Expand file tree Collapse file tree Original file line number Diff line number Diff 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:
You can’t perform that action at this time.
0 commit comments