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

luweizhou2016 · web-flow · commit 60881ec21f8f · 2022-06-09T11:19:12.000+08:00
diff --git 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<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
`@@ -1411,6 +1411,7 @@ void _jit_avx512_common_conv_bwd_data_kernel_f32<Vmm>::compute_loop_fma(`
`1411`	`1411`	`}`
`1412`	`1412`
`1413`	`1413`	`if (jcp.ndims == 5) {`
	`1414`	`+ base_post_ops_data_offset += reg64_size;`
`1414`	`1415`	`push(reg_src);`
`1415`	`1416`
`1416`	`1417`	`mov(reg_ki, ptr[param + GET_OFF(kd_padding)]);`