We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 19aaca9 commit 26da853Copy full SHA for 26da853
cuda_bindings/setup.py
@@ -240,8 +240,11 @@ def generate_output(infile, local):
240
] + include_path_list
241
library_dirs = [sysconfig.get_path("platlib"), os.path.join(os.sys.prefix, "lib")]
242
cudalib_subdirs = [r"lib\x64"] if sys.platform == "win32" else ["lib64", "lib"]
243
-library_dirs.extend(os.path.join(prefix, subdir) for prefix in CUDA_HOME
244
- for subdir in cudalib_subdirs)
+library_dirs.extend(
+ os.path.join(prefix, subdir)
245
+ for prefix in CUDA_HOME
246
+ for subdir in cudalib_subdirs
247
+)
248
249
extra_compile_args = []
250
extra_cythonize_kwargs = {}
0 commit comments