Skip to content

Commit 61e14a5

Browse files
committed
[CPU] [ARM] JIT GeluTanh emitter fix
1 parent 9a7045d commit 61e14a5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/plugins/intel_cpu/src/emitters/plugin/aarch64/jit_eltwise_emitters.cpp

+4-4
Original file line numberDiff line numberDiff line change
@@ -712,16 +712,16 @@ void jit_gelu_tanh_emitter::emit_isa(const std::vector<size_t> &in_vec_idxs, con
712712

713713
tanh_emitter->emit_code(
714714
{ vmm_aux0.getIdx() },
715-
out_vec_idxs,
715+
{ vmm_aux0.getIdx() },
716716
aux_vec_idxs,
717717
aux_gpr_idxs);
718718

719719
// compute 0.5 * x * (1 + tanh(G(x)))
720720
h->ld1r(vmm_aux1.s, table_val2("one"));
721-
h->fadd(vmm_dst.s, vmm_aux1.s, vmm_dst.s);
721+
h->fadd(vmm_aux0.s, vmm_aux1.s, vmm_aux0.s);
722722
h->ld1r(vmm_aux1.s, table_val2("half"));
723-
h->fmul(vmm_dst.s, vmm_aux1.s, vmm_dst.s);
724-
h->fmul(vmm_dst.s, store_src ? vmm_aux2.s : vmm_src.s, vmm_dst.s);
723+
h->fmul(vmm_aux0.s, vmm_aux1.s, vmm_aux0.s);
724+
h->fmul(vmm_dst.s, store_src ? vmm_aux2.s : vmm_src.s, vmm_aux0.s);
725725
}
726726

727727
void jit_gelu_tanh_emitter::register_table_entries() {

0 commit comments

Comments
 (0)