Merge pull request #218 from vzhurba01/patch-11.8.5

Update __pyx_capi__ due to layout change
NVIDIA · Nov 6, 2024 · 965a695 · 965a695
2 parents e10a4c4 + 74069a6
commit 965a695
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 4 deletions.
diff --git a/cuda_bindings/cuda/ccuda.pyx b/cuda_bindings/cuda/ccuda.pyx
@@ -5,3 +5,8 @@ cdef extern from *:
     #pragma message ( "The cuda.ccuda module is deprecated and will be removed in a future release, " \
                       "please switch to use the cuda.bindings.cydriver module instead." )
     """
+
+
+from cuda.bindings import cydriver
+__pyx_capi__ = cydriver.__pyx_capi__
+del cydriver
diff --git a/cuda_bindings/cuda/ccudart.pyx b/cuda_bindings/cuda/ccudart.pyx
@@ -5,3 +5,8 @@ cdef extern from *:
     #pragma message ( "The cuda.ccudart module is deprecated and will be removed in a future release, " \
                       "please switch to use the cuda.bindings.cyruntime module instead." )
     """
+
+
+from cuda.bindings import cyruntime
+__pyx_capi__ = cyruntime.__pyx_capi__
+del cyruntime
diff --git a/cuda_bindings/cuda/cnvrtc.pyx b/cuda_bindings/cuda/cnvrtc.pyx
@@ -5,3 +5,8 @@ cdef extern from *:
     #pragma message ( "The cuda.cnvrtc module is deprecated and will be removed in a future release, " \
                       "please switch to use the cuda.bindings.cynvrtc module instead." )
     """
+
+
+from cuda.bindings import cynvrtc
+__pyx_capi__ = cynvrtc.__pyx_capi__
+del cynvrtc
diff --git a/cuda_bindings/setup.py b/cuda_bindings/setup.py
@@ -228,8 +228,6 @@ def do_cythonize(extensions):
     ["cuda/bindings/*.pyx"],
     # public (deprecated, to be removed)
     ["cuda/*.pyx"],
-    # tests
-    ["tests/*.pyx"],
 ]
 
 for sources in sources_list:
@@ -260,9 +258,9 @@ def finalize_options(self):
 setup(
     version=versioneer.get_version(),
     ext_modules=do_cythonize(extensions),
-    packages=find_packages(include=["cuda.cuda", "cuda.cuda.*", "cuda.cuda.bindings", "cuda.cuda.bindings._bindings", "cuda.cuda.bindings._lib", "cuda.cuda.bindings._lib.cyruntime", "tests"]),
+    packages=find_packages(include=["cuda", "cuda.*", "cuda.bindings", "cuda.bindings._bindings", "cuda.bindings._lib", "cuda.bindings._lib.cyruntime"]),
     package_data=dict.fromkeys(
-        find_packages(include=["cuda.cuda", "cuda.cuda.*", "cuda.cuda.bindings", "cuda.cuda.bindings._bindings", "cuda.cuda.bindings._lib", "cuda.cuda.bindings._lib.cyruntime", "tests"]),
+        find_packages(include=["cuda", "cuda.*", "cuda.bindings", "cuda.bindings._bindings", "cuda.bindings._lib", "cuda.bindings._lib.cyruntime"]),
         ["*.pxd", "*.pyx", "*.py", "*.h", "*.cpp"],
     ),
     cmdclass=cmdclass,