Skip to content

Commit 1ffce8c

Browse files
committed
fix build error on noavx
test=develop
1 parent a2d9b34 commit 1ffce8c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

paddle/fluid/operators/math/jit_kernel_exp.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -269,6 +269,8 @@ REGISTER_JITKERNEL(vtanh, VTanhKernel);
269269

270270
namespace detail {
271271

272+
#ifdef __AVX__
273+
272274
#define ALIGN32 __attribute__((aligned(32)))
273275

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

402405
#ifdef __AVX2__
403406
__m256 ExpAVX2(__m256 x) {

0 commit comments

Comments
 (0)