Skip to content

Commit a0dadc5

Browse files
usstqazhai219
authored andcommitted
[FIX] CPU: x64: fix issue in eltwise post ops to allow multi-instance of same kind of alg
1 parent d3b15a9 commit a0dadc5

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/cpu/x64/injectors/jit_uni_postops_injector.cpp

+6-3
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,9 @@ jit_uni_postops_injector_t<isa, Vmm>::jit_uni_postops_injector_t(
6060
const auto &esp = eltwise_static_params;
6161
const auto &qsp = quantization_static_params;
6262

63-
for (const auto &post_op : post_ops.entry_) {
63+
for (int i = 0; i < post_ops_.len(); i++) {
64+
const auto &post_op = post_ops_.entry_[i];
65+
6466
if (post_op.is_eltwise()) {
6567
alg_to_eltwise_injector_.emplace(post_op.eltwise.alg,
6668
jit_uni_eltwise_injector<isa, Vmm>(host_, post_op.eltwise, data_type::f32,
@@ -98,8 +100,9 @@ jit_uni_postops_injector_t<isa, Vmm>::jit_uni_postops_injector_t(
98100
bool is_like_binary = false;
99101
bool is_eltwise = false;
100102

101-
for (int i = 0; i < post_ops.len(); i++) {
102-
const auto &post_op = post_ops.entry_[i];
103+
for (int i = 0; i < post_ops_.len(); i++) {
104+
const auto &post_op = post_ops_.entry_[i];
105+
103106
if (post_op.is_eltwise()) {
104107
is_eltwise = true;
105108
// Note: `dt` argument for eltwise injector is not propagated from

0 commit comments

Comments
 (0)