Attempted to disable vectorisation if __NVCOMPILER is defined.

ecmwf · Feb 12, 2024 · fc64ae4 · fc64ae4
1 parent 8e1484b
commit fc64ae4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/atlas/interpolation/method/sphericalvector/SparseMatrix.h b/src/atlas/interpolation/method/sphericalvector/SparseMatrix.h
@@ -13,6 +13,14 @@
 
 #include "atlas/library/defines.h"
 #if ATLAS_HAVE_EIGEN
+
+#ifdef __NVCOMPILER
+#define EIGEN_MAX_ALIGN_BYTES 0
+#define EIGEN_UNALIGNED_VECTORIZE 0
+#define EIGEN_DONT_VECTORIZE
+#define EIGEN_DONT_PARALLELIZE
+#endif
+
 #include <Eigen/Sparse>
 #endif