Merge pull request #430 from spotify/fix-402

fix avx 512 problem

setup.py

@@ -39,10 +39,6 @@
 extra_compile_args = []
 extra_link_args = []
 
-if os.environ.get('TRAVIS') == 'true':
-    # Resolving some annoying issue
-    extra_compile_args += ['-mno-avx']
-
 # Not all CPUs have march as a tuning parameter
 cputune = ['-march=native',]
 if platform.machine() == 'ppc64le':

src/annoylib.h

@@ -74,12 +74,14 @@ typedef unsigned __int64  uint64_t;
 #define popcount cole_popcount
 #endif
 
-#ifndef NO_MANUAL_VECTORIZATION
-#if defined(__AVX512F__)
+#if !defined(NO_MANUAL_VECTORIZATION) && defined(__GNUC__) && (__GNUC__ >6) && defined(__AVX512F__)  // See #402
+#pragma message "Using 512-bit AVX instructions"
 #define USE_AVX512
-#elif defined(__AVX__) && defined (__SSE__) && defined(__SSE2__) && defined(__SSE3__)
+#elif !defined(NO_MANUAL_VECTORIZATION) && defined(__AVX__) && defined (__SSE__) && defined(__SSE2__) && defined(__SSE3__)
+#pragma message "Using 128-bit AVX instructions"
 #define USE_AVX
-#endif
+#else
+#pragma message "Using no AVX instructions"
 #endif
 
 #if defined(USE_AVX) || defined(USE_AVX512)
@@ -989,6 +991,8 @@ template<typename S, typename T, typename Distance, typename Random>
       // Delete file if it already exists (See issue #335)
       unlink(filename);
 
+      printf("path: %s\n", filename);
+
       FILE *f = fopen(filename, "wb");
       if (f == NULL) {
         showUpdate("Unable to open: %s\n", strerror(errno));