From 36b0f69a689af851cf49002cba7e609cffac07be Mon Sep 17 00:00:00 2001 From: Luwei Zhou Date: Tue, 7 Jun 2022 16:04:09 +0800 Subject: [PATCH] Fix deconv 3D post OPs segment fault issue. --- src/cpu/x64/jit_avx512_common_conv_kernel.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/src/cpu/x64/jit_avx512_common_conv_kernel.cpp b/src/cpu/x64/jit_avx512_common_conv_kernel.cpp index 76d3fa7e159..0d7975cd952 100644 --- a/src/cpu/x64/jit_avx512_common_conv_kernel.cpp +++ b/src/cpu/x64/jit_avx512_common_conv_kernel.cpp @@ -1411,6 +1411,7 @@ void _jit_avx512_common_conv_bwd_data_kernel_f32::compute_loop_fma( } if (jcp.ndims == 5) { + base_post_ops_data_offset += reg64_size; push(reg_src); mov(reg_ki, ptr[param + GET_OFF(kd_padding)]);