updated single header

d6db78aa · Peter Steinbach · a4cb7b15 · d6db78aa
Commit d6db78aa authored Jun 08, 2018 by Peter Steinbach
Hide whitespace changes
Inline Side-by-side

Showing with 12 additions and 4 deletions

single_include/compass.hpp single_include/compass.hpp +12 -4

No files found.
--- a/single_include/compass.hpp
+++ b/single_include/compass.hpp
@@ -271,10 +271,18 @@ namespace compass {

 #else

-    #if _M_IX86_FP >= 2
-        #define COMPASS_CT_HAS_SSE2 1
-        #define COMPASS_CT_HAS_SSE3 1
-        #define COMPASS_CT_HAS_SSE4 1
+    #ifdef _M_IX86_FP
+		#if _M_IX86_FP >= 2
+			#define COMPASS_CT_HAS_SSE2 1
+			#define COMPASS_CT_HAS_SSE3 1
+			#define COMPASS_CT_HAS_SSE4 1
+		#endif
+	#else
+		#if defined(__AVX__) || defined(__AVX2__)
+			#define COMPASS_CT_HAS_SSE2 1
+			#define COMPASS_CT_HAS_SSE3 1
+			#define COMPASS_CT_HAS_SSE4 1
+		#endif
    #endif
 #endif
 #ifndef COMPASS_TAGS_H_