From d24d282a7ae585e25faab3c9e4252d586757f5dc Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Thu, 25 Oct 2018 14:13:36 +0800 Subject: [PATCH] fix avx error test=develop --- paddle/fluid/operators/math/jit_kernel_rnn.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddle/fluid/operators/math/jit_kernel_rnn.cc b/paddle/fluid/operators/math/jit_kernel_rnn.cc index c0847f0bee4..fab293f7d03 100644 --- a/paddle/fluid/operators/math/jit_kernel_rnn.cc +++ b/paddle/fluid/operators/math/jit_kernel_rnn.cc @@ -136,6 +136,7 @@ static std::shared_ptr> GetActKernel( return nullptr; } +#ifdef __AVX__ template static std::unique_ptr GetAVXAct(const std::string& type) { if (type == "sigmoid") { @@ -150,6 +151,7 @@ static std::unique_ptr GetAVXAct(const std::string& type) { PADDLE_THROW("Not support type: %s", type); return nullptr; } +#endif /* LSTM JitKernel */ template -- GitLab