fix gru gcc7.4 bug for the gru compile

fix gru gcc7.4 bug for the gru compile
musl/disable_test_yolov3_temporarily
Jack Zhou 5 years ago committed by GitHub
parent 1476e1f998
commit bc6033f86b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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> {
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
} }
}; };

Loading…
Cancel
Save