Fix deconv 3D post OPs segment fault issue. (#130)

luweizhou2016 · EgorDuplensky · commit 8a4ce9446ffd · 2023-03-03T11:59:05.000+01:00
diff --git a/src/cpu/x64/jit_avx512_common_conv_kernel.cpp b/src/cpu/x64/jit_avx512_common_conv_kernel.cpp
@@ -1414,6 +1414,7 @@ void _jit_avx512_common_conv_bwd_data_kernel_f32<Vmm>::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)]);

Original file line number	Diff line number	Diff line change
`@@ -1414,6 +1414,7 @@ void _jit_avx512_common_conv_bwd_data_kernel_f32<Vmm>::compute_loop_fma(`
`1414`	`1414`	`}`
`1415`	`1415`
`1416`	`1416`	`if (jcp.ndims == 5) {`
	`1417`	`+ base_post_ops_data_offset += reg64_size;`
`1417`	`1418`	`push(reg_src);`
`1418`	`1419`
`1419`	`1420`	`mov(reg_ki, ptr[param + GET_OFF(kd_padding)]);`