Explicitly state which extension uses cpp files

Each cpp file is only used by a single extension
NVIDIA · Jan 7, 2025 · 8620a28 · 8620a28
1 parent 7ffb964
commit 8620a28
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/cuda_bindings/setup.py b/cuda_bindings/setup.py
@@ -264,10 +264,12 @@ def do_cythonize(extensions):
 
 sources_list = [
     # private
-    ["cuda/bindings/_bindings/*.pyx", "cuda/bindings/_bindings/loader.cpp"],
+    ["cuda/bindings/_bindings/cydriver.pyx", "cuda/bindings/_bindings/loader.cpp"],
+    ["cuda/bindings/_bindings/cynvrtc.pyx"],
     # utils
-    ["cuda/bindings/_lib/*.pyx", "cuda/bindings/_lib/param_packer.cpp"],
-    ["cuda/bindings/_lib/cyruntime/*.pyx"],
+    ["cuda/bindings/_lib/utils.pyx", "cuda/bindings/_lib/param_packer.cpp"],
+    ["cuda/bindings/_lib/cyruntime/cyruntime.pyx"],
+    ["cuda/bindings/_lib/cyruntime/utils.pyx"],
     # public
     ["cuda/bindings/*.pyx"],
     # public (deprecated, to be removed)