detect clang CUDA too in math_helper.h

lattice · Jan 25, 2024 · ccbba91 · ccbba91
1 parent 4844105
commit ccbba91
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/targets/cuda/math_helper.h b/include/targets/cuda/math_helper.h
@@ -1,4 +1,4 @@
-#if defined(__NVCC__)
+#if defined(__CUDACC__)
 
 #include <math_helper.cuh>