fix ci build

fangq · Aug 28, 2023 · 2a5f27c · 2a5f27c
1 parent b9f22ad
commit 2a5f27c
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/pymcx b/pymcx
diff --git a/src/mcx_core.cu b/src/mcx_core.cu
@@ -85,6 +85,14 @@ This unit is written with CUDA-C and shall be compiled using nvcc in cuda-toolki
 #define FL3(f) make_float3(f,f,f)
 #define CUDA_NAN_F    (__int_as_float(0x7fffffff)) /**< NaN constant in CUDA */
 
+#if !defined(__CUDA_ARCH_LIST__)
+    #if defined(__CUDA_ARCH__)
+        #define __CUDA_ARCH_LIST__ __CUDA_ARCH__
+    #else
+        #define __CUDA_ARCH_LIST__ 350
+    #endif
+#endif
+
 /**
  * @brief Adding two float3 vectors c=a+b
  */

diff --git a/src/pybind11 b/src/pybind11