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

luweizhou2016 · luweizhou2016 · commit a04871025bdc · 2023-05-25T02:57:26.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
@@ -1415,6 +1415,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
`@@ -1415,6 +1415,7 @@ void _jit_avx512_common_conv_bwd_data_kernel_f32<Vmm>::compute_loop_fma(`
`1415`	`1415`	`}`
`1416`	`1416`
`1417`	`1417`	`if (jcp.ndims == 5) {`
	`1418`	`+ base_post_ops_data_offset += reg64_size;`
`1418`	`1419`	`push(reg_src);`
`1419`	`1420`
`1420`	`1421`	`mov(reg_ki, ptr[param + GET_OFF(kd_padding)]);`