Skip to content

Commit 77dbde4

Browse files
committed
fix
1 parent 571f60b commit 77dbde4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cuda_bindings/setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,7 @@ def do_cythonize(extensions):
343343
(["cuda/*.pyx"], None),
344344
# internal files used by generated bindings
345345
(["cuda/bindings/_internal/utils.pyx"], None),
346-
*(([f], None) for f in dst_files),
346+
*(([f], None) for f in dst_files if f.endswith(".pyx")),
347347
]
348348
if sys.platform == "win32":
349349
# cuFILE does not support Windows
@@ -352,7 +352,7 @@ def do_cythonize(extensions):
352352
file_list, _ = source
353353
if all("cufile" not in f for f in file_list):
354354
new_sources_list.append(source)
355-
assert len(new_sources_list) == len(sources_list) - 3
355+
assert len(new_sources_list) == len(sources_list) - 3, f"{new_sources_list=}, {sources_list=}"
356356
sources_list = new_sources_list
357357

358358
for sources, libraries in sources_list:

0 commit comments

Comments
 (0)