Skip to content

Commit 26da853

Browse files
committed
Formatting fix
1 parent 19aaca9 commit 26da853

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

cuda_bindings/setup.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -240,8 +240,11 @@ def generate_output(infile, local):
240240
] + include_path_list
241241
library_dirs = [sysconfig.get_path("platlib"), os.path.join(os.sys.prefix, "lib")]
242242
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)
243+
library_dirs.extend(
244+
os.path.join(prefix, subdir)
245+
for prefix in CUDA_HOME
246+
for subdir in cudalib_subdirs
247+
)
245248

246249
extra_compile_args = []
247250
extra_cythonize_kwargs = {}

0 commit comments

Comments
 (0)