Add macro for loop unrolling across compilers

AMReX-Codes · Jan 16, 2024 · 8c185ca · 8c185ca
1 parent 03f5936
commit 8c185ca
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/Src/Base/AMReX_Extension.H b/Src/Base/AMReX_Extension.H
@@ -133,6 +133,21 @@
 #define AMREX_NO_INLINE
 #endif
 
+// unroll loop
+#define AMREX_TO_STRING_HELPER(X)   #X
+#define AMREX_TO_STRING(X)          AMREX_TO_STRING_HELPER(X)
+
+#if defined(__GNUC__)
+#define AMREX_UNROLL_LOOP(n)  _Pragma(AMREX_TO_STRING(GCC unroll (n)))
+#elif defined(__clang__) || defined(__CUDACC__) || defined(__HIP__) || defined(__INTEL_CLANG_COMPILER))
+#define AMREX_UNROLL_LOOP(n)  _Pragma(AMREX_TO_STRING(unroll (n)))
+#else
+#define AMREX_UNROLL_LOOP(n)
+#endif
+
+//#undef AMREX_TO_STRING_HELPER
+//#undef AMREX_TO_STRING
+
 // __attribute__((weak))
 
 #if defined(AMREX_TYPECHECK)