conditionally use newer nvcc diagnostic supression pragma

alpaka-group · Oct 30, 2021 · 66d47eb · 66d47eb
1 parent 41a96b8
commit 66d47eb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/include/llama/llama.hpp b/include/llama/llama.hpp
@@ -28,7 +28,11 @@
 
 #ifdef __NVCC__
 #    pragma push
-#    pragma diag_suppress 940
+#    if __CUDACC_VER_MAJOR__ * 1000 + __CUDACC_VER_MINOR__ >= 11005
+#        pragma nv_diag_suppress 940
+#    else
+#        pragma diag_suppress 940
+#    endif
 #endif
 
 #include "ArrayExtents.hpp"