Skip to content

Commit 327b067

Browse files
committed
TPP
1 parent bec3062 commit 327b067

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/plugins/intel_cpu/src/transformations/tpp/x64/pass/eltwise_to_eltwise_tpp.cpp

+4
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,10 @@ EltwiseToEltwiseTPP::EltwiseToEltwiseTPP() {
3939
const size_t N_block =
4040
ov::is_type<ov::snippets::op::ReduceBase>(node) ? ov::snippets::utils::get_full_dim_value() : 64;
4141
ov::replace_node_update_name(node, tpp_eltwise);
42+
auto subtensor = {M_block, N_block};
43+
if (tpp_eltwise->input(0).get_partial_shape().size() == 1) {
44+
subtensor = {N_block};
45+
}
4246
for (size_t i = 0; i < node->get_input_size(); i++) {
4347
ov::snippets::lowered::PortDescriptorUtils::set_port_descriptor(tpp_eltwise->input(i), {M_block, N_block});
4448
}

0 commit comments

Comments
 (0)