ensure using c++

NVIDIA · Jan 10, 2025 · d32a05c · d32a05c
1 parent 98e3d78
commit d32a05c
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/cuda_bindings/tests_cython/test_ccuda.pyx b/cuda_bindings/tests_cython/test_ccuda.pyx
@@ -5,6 +5,9 @@
 # this software. Any use, reproduction, disclosure, or distribution of
 # this software and related documentation outside the terms of the EULA
 # is strictly prohibited.
+
+# distutils: language=c++
+
 from libc.string cimport (
     memset,
     memcmp
@@ -53,4 +56,4 @@ def test_ccuda_memcpy():
     err = ccuda.cuMemFree(dptr)
     assert(err == 0)
     err = ccuda.cuCtxDestroy(ctx)
-    assert(err == 0)
+    assert(err == 0)
diff --git a/cuda_bindings/tests_cython/test_ccudart.pyx b/cuda_bindings/tests_cython/test_ccudart.pyx
@@ -5,6 +5,9 @@
 # this software. Any use, reproduction, disclosure, or distribution of
 # this software and related documentation outside the terms of the EULA
 # is strictly prohibited.
+
+# distutils: language=c++
+
 from libc.string cimport (
     memset,
     memcmp

diff --git a/cuda_bindings/tests_cython/test_interoperability_cython.pyx b/cuda_bindings/tests_cython/test_interoperability_cython.pyx
@@ -5,6 +5,9 @@
 # this software. Any use, reproduction, disclosure, or distribution of
 # this software and related documentation outside the terms of the EULA
 # is strictly prohibited.
+
+# distutils: language=c++
+
 import pytest
 import cuda.cuda as cuda
 import cuda.cudart as cudart