@@ -18,8 +18,8 @@ def _no_such_file_in_sub_dirs(sub_dirs, file_wild, error_messages, attachments):
18
18
attachments .append (f" { node } " )
19
19
20
20
21
- def _find_so_using_nvidia_lib_dirs (libbasename , so_basename , error_messages , attachments ):
22
- if libbasename == "nvvm" : # noqa: SIM108
21
+ def _find_so_using_nvidia_lib_dirs (libname , so_basename , error_messages , attachments ):
22
+ if libname == "nvvm" : # noqa: SIM108
23
23
nvidia_sub_dirs = ("nvidia" , "*" , "nvvm" , "lib64" )
24
24
else :
25
25
nvidia_sub_dirs = ("nvidia" , "*" , "lib" )
@@ -39,12 +39,12 @@ def _find_so_using_nvidia_lib_dirs(libbasename, so_basename, error_messages, att
39
39
return None
40
40
41
41
42
- def _find_dll_using_nvidia_bin_dirs (libbasename , error_messages , attachments ):
43
- if libbasename == "nvvm" : # noqa: SIM108
42
+ def _find_dll_using_nvidia_bin_dirs (libname , error_messages , attachments ):
43
+ if libname == "nvvm" : # noqa: SIM108
44
44
nvidia_sub_dirs = ("nvidia" , "*" , "nvvm" , "bin" )
45
45
else :
46
46
nvidia_sub_dirs = ("nvidia" , "*" , "bin" )
47
- file_wild = libbasename + "*.dll"
47
+ file_wild = libname + "*.dll"
48
48
for bin_dir in sys_path_find_sub_dirs (nvidia_sub_dirs ):
49
49
for node in sorted (glob .glob (os .path .join (bin_dir , file_wild ))):
50
50
dll_name = os .path .join (bin_dir , node )
@@ -93,11 +93,11 @@ def _find_so_using_cudalib_dir(so_basename, error_messages, attachments):
93
93
return None
94
94
95
95
96
- def _find_dll_using_cudalib_dir (libbasename , error_messages , attachments ):
96
+ def _find_dll_using_cudalib_dir (libname , error_messages , attachments ):
97
97
cudalib_dir = _get_cuda_paths_info ("cudalib_dir" , error_messages )
98
98
if cudalib_dir is None :
99
99
return None
100
- file_wild = libbasename + "*.dll"
100
+ file_wild = libname + "*.dll"
101
101
for node in sorted (glob .glob (os .path .join (cudalib_dir , file_wild ))):
102
102
dll_name = os .path .join (cudalib_dir , node )
103
103
if os .path .isfile (dll_name ):
@@ -110,26 +110,26 @@ def _find_dll_using_cudalib_dir(libbasename, error_messages, attachments):
110
110
111
111
112
112
@functools .cache
113
- def find_nvidia_dynamic_library (libbasename ) :
113
+ def find_nvidia_dynamic_library (name : str ) -> str :
114
114
error_messages = []
115
115
attachments = []
116
116
117
117
if IS_WIN32 :
118
- dll_name = _find_dll_using_nvidia_bin_dirs (libbasename , error_messages , attachments )
118
+ dll_name = _find_dll_using_nvidia_bin_dirs (name , error_messages , attachments )
119
119
if dll_name is None :
120
- if libbasename == "nvvm" :
120
+ if name == "nvvm" :
121
121
dll_name = _get_cuda_paths_info ("nvvm" , error_messages )
122
122
else :
123
- dll_name = _find_dll_using_cudalib_dir (libbasename , error_messages , attachments )
123
+ dll_name = _find_dll_using_cudalib_dir (name , error_messages , attachments )
124
124
if dll_name is None :
125
125
attachments = "\n " .join (attachments )
126
- raise RuntimeError (f"Failure finding { libbasename } *.dll: { ', ' .join (error_messages )} \n { attachments } " )
126
+ raise RuntimeError (f"Failure finding { name } *.dll: { ', ' .join (error_messages )} \n { attachments } " )
127
127
return dll_name
128
128
129
- so_basename = f"lib{ libbasename } .so"
130
- so_name = _find_so_using_nvidia_lib_dirs (libbasename , so_basename , error_messages , attachments )
129
+ so_basename = f"lib{ name } .so"
130
+ so_name = _find_so_using_nvidia_lib_dirs (name , so_basename , error_messages , attachments )
131
131
if so_name is None :
132
- if libbasename == "nvvm" :
132
+ if name == "nvvm" :
133
133
so_name = _get_cuda_paths_info ("nvvm" , error_messages )
134
134
else :
135
135
so_name = _find_so_using_cudalib_dir (so_basename , error_messages , attachments )
0 commit comments