未验证 提交 bc6033f8 编写于 作者: J Jack Zhou 提交者: GitHub

fix gru gcc7.4 bug for the gru compile

fix gru gcc7.4 bug for the gru compile
上级 1476e1f9
...@@ -43,7 +43,7 @@ struct GRUUnitFunctor<platform::CPUDeviceContext, T> { ...@@ -43,7 +43,7 @@ struct GRUUnitFunctor<platform::CPUDeviceContext, T> {
detail::forward_reset_output(detail::forward::gru_resetOutput<T>(), value, detail::forward_reset_output(detail::forward::gru_resetOutput<T>(), value,
frame_size, batch_size, active_gate, true, frame_size, batch_size, active_gate, true,
&context); nullptr);
if (value.prev_out_value) { if (value.prev_out_value) {
blas.GEMM(false, false, batch_size, frame_size, frame_size, 1, blas.GEMM(false, false, batch_size, frame_size, frame_size, 1,
...@@ -54,7 +54,7 @@ struct GRUUnitFunctor<platform::CPUDeviceContext, T> { ...@@ -54,7 +54,7 @@ struct GRUUnitFunctor<platform::CPUDeviceContext, T> {
detail::forward_final_output(detail::forward::gru_finalOutput<T>(), value, detail::forward_final_output(detail::forward::gru_finalOutput<T>(), value,
frame_size, batch_size, active_node, frame_size, batch_size, active_node,
origin_mode, &context); origin_mode, true, nullptr);
#endif #endif
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册