add the jit support, test=develop

6 years ago · 5e46c98362
parent c75dc885b5
commit 5e46c98362
3 changed files with 10 additions and 6 deletions
--- 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"

--- 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 {
--- 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__