fix build error on noavx

test=develop
6 years ago · 1ffce8c0ae
parent a2d9b34417
commit 1ffce8c0ae
1 changed files with 3 additions and 0 deletions
--- a/paddle/fluid/operators/math/jit_kernel_exp.cc
+++ b/paddle/fluid/operators/math/jit_kernel_exp.cc
@ -269,6 +269,8 @@ REGISTER_JITKERNEL(vtanh, VTanhKernel);

 namespace detail {

+#ifdef __AVX__
+
 #define ALIGN32 __attribute__((aligned(32)))

 #define _PS256_CONST(Name, Val)                                      \
@ -398,6 +400,7 @@ __m256 ExpAVX(__m256 x) {
  y = _mm256_mul_ps(y, pow2n);
  return y;
 }
+#endif

 #ifdef __AVX2__
 __m256 ExpAVX2(__m256 x) {