Fix gru performace decline in 1.8.5 (#29455)

revert-31562-mean
Jack Zhou 5 years ago committed by GitHub
parent acce962133
commit c7cada8571
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -276,7 +276,7 @@ inline void forward_reset_output(
// use eigen
forward_reset_outputV2(*context, value, frame_size);
} else {
if (OpResetOutput::avx && (frame_size & static_cast<int>(8 - 1)) &&
if (OpResetOutput::avx && (frame_size > static_cast<int>(8 - 1)) &&
(sizeof(T) == 4)) {
hl_avx_gru_forward_reset_output(
op_reset_output, value.gate_value, value.reset_output_value,
@ -329,7 +329,7 @@ inline void forward_final_output(
// eigen
forward_final_outputV2(*context, value, frame_size);
} else {
if (OpFinalOutput::avx && (frame_size & static_cast<int>(8 - 1)) &&
if (OpFinalOutput::avx && (frame_size > static_cast<int>(8 - 1)) &&
(sizeof(T) == 4)) {
hl_avx_gru_forward_final_output(op_final_output, value.gate_value,
value.prev_out_value,

Loading…
Cancel
Save