Check __AVX512__ macro

kimwalisch · Apr 4, 2024 · 0e9ca8a · 0e9ca8a
1 parent 4ee88fb
commit 0e9ca8a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 12 deletions.
diff --git a/include/primesieve/PrimeGenerator.hpp b/include/primesieve/PrimeGenerator.hpp
@@ -24,14 +24,13 @@
 #include <cstddef>
 
 #if defined(MULTIARCH_AVX512)
-  #if defined(__AVX512__) || \
-      (defined(__AVX512F__) && \
-       defined(__AVX512VBMI__) && \
-       defined(__AVX512VBMI2__))
-    // GCC/Clang function multiversioning for AVX512 is not needed if
-    // the user compiles with -mavx512f -mavx512vbmi -mavx512vbmi2.
-    // GCC/Clang function multiversioning generally causes a minor
-    // overhead, hence we disable it if it is not needed.
+  // GCC/Clang function multiversioning for AVX512 is not needed if
+  // the user compiles with -mavx512f -mavx512vbmi -mavx512vbmi2.
+  // GCC/Clang function multiversioning generally causes a minor
+  // overhead, hence we disable it if it is not needed.
+  #if defined(__AVX512__) || (defined(__AVX512F__) && \
+                              defined(__AVX512VBMI__) && \
+                              defined(__AVX512VBMI2__))
     #undef MULTIARCH_AVX512
   #else
     #define MULTIARCH_TARGET_DEFAULT

diff --git a/src/PrimeGenerator.cpp b/src/PrimeGenerator.cpp
@@ -37,10 +37,9 @@
 
 // x86-64 AVX512
 #if __has_include(<immintrin.h>) && \
-    (defined(__AVX512__) || \
-       (defined(__AVX512F__) && \
-        defined(__AVX512VBMI__) && \
-        defined(__AVX512VBMI2__)))
+   (defined(__AVX512__) || (defined(__AVX512F__) && \
+                            defined(__AVX512VBMI__) && \
+                            defined(__AVX512VBMI2__)))
   #include <immintrin.h>
   #define HAS_AVX512_VBMI2