Skip to content

Commit 9de81ff

Browse files
committed
cpu: aarch64: fix guard on bwd specific logic in eltwise
1 parent 47573db commit 9de81ff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/cpu/aarch64/jit_uni_eltwise.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ struct jit_uni_kernel_t : public jit_uni_eltwise_kernel_t {
216216

217217
add_imm(reg_src, reg_src, dtype_size(), X_TMP_0);
218218
add_imm(reg_dst, reg_dst, dtype_size(), X_TMP_0);
219-
add_imm(reg_diff_dst, reg_diff_dst, dtype_size(), X_TMP_0);
219+
if (!is_fwd) add_imm(reg_diff_dst, reg_diff_dst, dtype_size(), X_TMP_0);
220220
subs(reg_work_amount, reg_work_amount, 1);
221221

222222
b(remainder_loop_start);

0 commit comments

Comments
 (0)