提交 1ffce8c0 编写于 作者: T tensor-tang

fix build error on noavx

test=develop
上级 a2d9b344
...@@ -269,6 +269,8 @@ REGISTER_JITKERNEL(vtanh, VTanhKernel); ...@@ -269,6 +269,8 @@ REGISTER_JITKERNEL(vtanh, VTanhKernel);
namespace detail { namespace detail {
#ifdef __AVX__
#define ALIGN32 __attribute__((aligned(32))) #define ALIGN32 __attribute__((aligned(32)))
#define _PS256_CONST(Name, Val) \ #define _PS256_CONST(Name, Val) \
...@@ -398,6 +400,7 @@ __m256 ExpAVX(__m256 x) { ...@@ -398,6 +400,7 @@ __m256 ExpAVX(__m256 x) {
y = _mm256_mul_ps(y, pow2n); y = _mm256_mul_ps(y, pow2n);
return y; return y;
} }
#endif
#ifdef __AVX2__ #ifdef __AVX2__
__m256 ExpAVX2(__m256 x) { __m256 ExpAVX2(__m256 x) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册