fix build error on noavx

test=develop

fix build error on noavx
test=develop
1ffce8c0 · tensor-tang · a2d9b344 · 1ffce8c0
显示空白变更内容
内联并排

Showing with 3 addition and 0 deletion

paddle/fluid/operators/math/jit_kernel_exp.cc paddle/fluid/operators/math/jit_kernel_exp.cc +3 -0

未找到文件。
--- 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) {