Guard sleef for AVX/AVX2 (pytorch#7678)

cpuhrsch · web-flow · commit 431c80a1289c · 2018-05-18T17:33:21.000Z
diff --git a/aten/src/ATen/cpu/vec256/vec256_double.h b/aten/src/ATen/cpu/vec256/vec256_double.h
@@ -2,8 +2,9 @@
 
 #include "intrinsics.h"
 #include "vec256_base.h"
+#ifdef __AVX__
 #include <sleef.h>
-#include <iostream>
+#endif
 
 namespace at {
 namespace vec256 {
diff --git a/aten/src/ATen/cpu/vec256/vec256_float.h b/aten/src/ATen/cpu/vec256/vec256_float.h
@@ -2,7 +2,9 @@
 
 #include "intrinsics.h"
 #include "vec256_base.h"
+#ifdef __AVX__
 #include <sleef.h>
+#endif
 
 namespace at {
 namespace vec256 {
