Skip to content

Commit

Permalink
cpu: x64: jit_avx2_1x1_conv: always reserve stack space
Browse files Browse the repository at this point in the history
  • Loading branch information
kwiersch authored and vpirogov committed Nov 18, 2022
1 parent 844326b commit b60633f
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions src/cpu/x64/jit_avx2_1x1_conv_kernel_f32.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -563,8 +563,7 @@ void jit_avx2_1x1_conv_kernel_f32::generate_diff_bias_loop(int load_loop_blk) {
void jit_avx2_1x1_conv_kernel_f32::generate() {
preamble();

if (jcp.with_binary || (jcp.with_bias && jcp.prop_kind == backward_weights))
sub(rsp, stack_space_needed);
sub(rsp, stack_space_needed);

if (jcp.with_binary) {
const auto zeroed_reg = r15;
Expand Down Expand Up @@ -668,8 +667,7 @@ void jit_avx2_1x1_conv_kernel_f32::generate() {

L(load_loop_blk_end);

if (jcp.with_binary || (jcp.with_bias && jcp.prop_kind == backward_weights))
add(rsp, stack_space_needed);
add(rsp, stack_space_needed);

postamble();

Expand Down

0 comments on commit b60633f

Please sign in to comment.