Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2299
Star 20931
Fork 5422
P
Paddle
提交 d24d282a 编写于 作者: T tensor-tang
浏览文件
操作

fix avx error 

test=develop
上级 9cb8738f
隐藏空白更改
内联 并排
Showing with 2 addition and 0 deletion
paddle/fluid/operators/math/jit_kernel_rnn.cc
浏览文件 @ d24d282a
...@@ -136,6 +136,7 @@ static std::shared_ptr<const VActKernel<T>> GetActKernel( ...@@ -136,6 +136,7 @@ static std::shared_ptr<const VActKernel<T>> GetActKernel(
return nullptr; return nullptr;
} }
#ifdef __AVX__
template <jit::cpu_isa_t isa> template <jit::cpu_isa_t isa>
static std::unique_ptr<AVXAct> GetAVXAct(const std::string& type) { static std::unique_ptr<AVXAct> GetAVXAct(const std::string& type) {
if (type == "sigmoid") { if (type == "sigmoid") {
...@@ -150,6 +151,7 @@ static std::unique_ptr<AVXAct> GetAVXAct(const std::string& type) { ...@@ -150,6 +151,7 @@ static std::unique_ptr<AVXAct> GetAVXAct(const std::string& type) {
PADDLE_THROW("Not support type: %s", type); PADDLE_THROW("Not support type: %s", type);
return nullptr; return nullptr;
} }
#endif
/* LSTM JitKernel */ /* LSTM JitKernel */
template <typename T, jit::cpu_isa_t isa, jit_block> template <typename T, jit::cpu_isa_t isa, jit_block>
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或