fix avx error

test=develop
6 years ago · d24d282a7a
parent 9cb8738f54
commit d24d282a7a
1 changed files with 2 additions and 0 deletions
--- 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<const VActKernel<T>> GetActKernel(
  return nullptr;
 }

+#ifdef __AVX__
 template <jit::cpu_isa_t isa>
 static std::unique_ptr<AVXAct> GetAVXAct(const std::string& type) {
  if (type == "sigmoid") {
@ -150,6 +151,7 @@ static std::unique_ptr<AVXAct> GetAVXAct(const std::string& type) {
  PADDLE_THROW("Not support type: %s", type);
  return nullptr;
 }
+#endif

 /* LSTM JitKernel */
 template <typename T, jit::cpu_isa_t isa, jit_block>