Skip to content

Commit d24d282

Browse files
committed
fix avx error
test=develop
1 parent 9cb8738 commit d24d282

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddle/fluid/operators/math/jit_kernel_rnn.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,7 @@ static std::shared_ptr<const VActKernel<T>> GetActKernel(
136136
return nullptr;
137137
}
138138

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

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

0 commit comments

Comments
 (0)