[FORK][FIX] jit_uni_dw_conv_kernel_f32: fixed register conflict

mangguo321 · mangguo321 · commit 4e29b771fcdf · 2024-02-22T10:53:36.000-05:00
diff --git a/src/cpu/x64/jit_uni_dw_conv_kernel_f32.cpp b/src/cpu/x64/jit_uni_dw_conv_kernel_f32.cpp
@@ -844,6 +844,8 @@ void jit_uni_dw_conv_bwd_data_kernel_f32<isa>::apply_postprocess(int ur_ch_block
     const auto &p = attr_.post_ops_;
     std::size_t post_ops_data_offset = 0;
     int depthwise_inj_idx = 0;
+    base_post_ops_data_offset += reg64_size;
+    push(reg_d_weights);
     for (int i = 0; i < p.len(); i++) {
         auto& post_op = p.entry_[i];
         if (post_op.is_depthwise()) {
@@ -864,6 +866,8 @@ void jit_uni_dw_conv_bwd_data_kernel_f32<isa>::apply_postprocess(int ur_ch_block
             depthwise_inj_idx++;
         }
     }
+    pop(reg_d_weights);
+    base_post_ops_data_offset -= reg64_size;
 }
 
 template <cpu_isa_t isa>

Original file line number	Diff line number	Diff line change
`@@ -844,6 +844,8 @@ void jit_uni_dw_conv_bwd_data_kernel_f32<isa>::apply_postprocess(int ur_ch_block`
`844`	`844`	`const auto &p = attr_.post_ops_;`
`845`	`845`	`std::size_t post_ops_data_offset = 0;`
`846`	`846`	`int depthwise_inj_idx = 0;`
	`847`	`+ base_post_ops_data_offset += reg64_size;`
	`848`	`+ push(reg_d_weights);`
`847`	`849`	`for (int i = 0; i < p.len(); i++) {`
`848`	`850`	`auto& post_op = p.entry_[i];`
`849`	`851`	`if (post_op.is_depthwise()) {`
`@@ -864,6 +866,8 @@ void jit_uni_dw_conv_bwd_data_kernel_f32<isa>::apply_postprocess(int ur_ch_block`
`864`	`866`	`depthwise_inj_idx++;`
`865`	`867`	`}`
`866`	`868`	`}`
	`869`	`+ pop(reg_d_weights);`
	`870`	`+ base_post_ops_data_offset -= reg64_size;`
`867`	`871`	`}`
`868`	`872`
`869`	`873`	`template <cpu_isa_t isa>`