Fix warning reenabling in llama.hpp

alpaka-group · Dec 14, 2022 · f9dc6ba · f9dc6ba
1 parent 6b300a2
commit f9dc6ba
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/include/llama/llama.hpp b/include/llama/llama.hpp
@@ -36,13 +36,15 @@
 #        pragma nv_diag_suppress 940
 #    else
 #        pragma diag_suppress 940
+
 #    endif
 #endif
 #ifdef __NVCOMPILER
 #    pragma push
 #    pragma diag_suppress 941
 #endif
 
+
 #include "ArrayExtents.hpp"
 #include "ArrayIndexRange.hpp"
 #include "BlobAllocators.hpp"
@@ -75,10 +77,14 @@
 #include "mapping/Trace.hpp"
 #include "mapping/tree/Mapping.hpp"
 
-#if defined(__CUDACC__) || defined(__NVCOMPILER)
+#if defined(__CUDACC__)
 #    ifdef __NVCC_DIAG_PRAGMA_SUPPORT__
 #        pragma nv_diag_default 940
 #    else
 #        pragma diag_default 940
 #    endif
 #endif
+#ifdef __NVCOMPILER
+#    pragma push
+#    pragma diag_default 941
+#endif