From 5e46c98362897fbf043eb8387618740fbbd6fd07 Mon Sep 17 00:00:00 2001
From: peizhilin <wopeizl@163.com>
Date: Sat, 17 Nov 2018 14:41:29 +0800
Subject: [PATCH] add the jit support, test=develop

---
 paddle/fluid/operators/math/detail/activation_functions.h | 6 ------
 paddle/fluid/platform/cpu_info.h                          | 5 +++++
 paddle/fluid/platform/enforce.h                           | 5 +++++
 3 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/paddle/fluid/operators/math/detail/activation_functions.h b/paddle/fluid/operators/math/detail/activation_functions.h
index 42fb45a8a..2b3d38d95 100644
--- a/paddle/fluid/operators/math/detail/activation_functions.h
+++ b/paddle/fluid/operators/math/detail/activation_functions.h
@@ -16,12 +16,6 @@ limitations under the License. */
 #include <math.h>
 #include <string>
 
-#ifdef _WIN32
-#undef __AVX__
-#undef __AVX__2
-#endif
-
-
 #include "paddle/fluid/platform/enforce.h"
 #include "paddle/fluid/platform/hostdevice.h"
 
diff --git a/paddle/fluid/platform/cpu_info.h b/paddle/fluid/platform/cpu_info.h
index 6810a1651..1b4840d9a 100644
--- a/paddle/fluid/platform/cpu_info.h
+++ b/paddle/fluid/platform/cpu_info.h
@@ -14,6 +14,11 @@ limitations under the License. */
 
 #pragma once
 
+#ifdef _WIN32
+#undef __AVX__
+#undef __AVX2__
+#endif
+
 #include <stddef.h>
 
 namespace paddle {
diff --git a/paddle/fluid/platform/enforce.h b/paddle/fluid/platform/enforce.h
index a251bfcd9..c03bbd59a 100644
--- a/paddle/fluid/platform/enforce.h
+++ b/paddle/fluid/platform/enforce.h
@@ -14,6 +14,11 @@ limitations under the License. */
 
 #pragma once
 
+#ifdef _WIN32
+#undef __AVX__
+#undef __AVX2__
+#endif
+
 #ifdef __GNUC__
 #include <cxxabi.h>  // for __cxa_demangle
 #endif               // __GNUC__
-- 
GitLab