TPP

Author: aobolensk

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

@@ -40,10 +40,18 @@ EltwiseToEltwiseTPP::EltwiseToEltwiseTPP() {
             ov::is_type<ov::snippets::op::ReduceBase>(node) ? ov::snippets::utils::get_full_dim_value() : 64;
         ov::replace_node_update_name(node, tpp_eltwise);
         for (size_t i = 0; i < node->get_input_size(); i++) {
-            ov::snippets::lowered::PortDescriptorUtils::set_port_descriptor(tpp_eltwise->input(i), {M_block, N_block});
+            auto subtensor = snippets::VectorDims{M_block, N_block};
+            if (tpp_eltwise->input(i).get_partial_shape().size() < 2) {
+                subtensor = snippets::VectorDims{N_block};
+            }
+            ov::snippets::lowered::PortDescriptorUtils::set_port_descriptor(tpp_eltwise->input(i), subtensor);
         }
 
-        ov::snippets::lowered::PortDescriptorUtils::set_port_descriptor(tpp_eltwise->output(0), {M_block, N_block});
+        auto subtensor = snippets::VectorDims{M_block, N_block};
+        if (tpp_eltwise->output(0).get_partial_shape().size() < 2) {
+            subtensor = snippets::VectorDims{N_block};
+        }
+        ov::snippets::lowered::PortDescriptorUtils::set_port_descriptor(tpp_eltwise->output(0), subtensor);
 
         return true;
     };

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

@@ -82,7 +82,15 @@ bool FuseTPPToEquations::fuse_from_root(const NodePtr& root, const std::shared_p
         kv.second = equation;
     replace_nodes(m, {}, node_replace_map);
     for (const auto& in : equation->inputs()) {
-        ov::snippets::lowered::PortDescriptorUtils::set_port_descriptor(in, root_subtensor);
+        auto subtensor = root_subtensor;
+        if (in.get_partial_shape().size() < 2) {
+            subtensor = {root_subtensor.back()};
+        }
+        ov::snippets::lowered::PortDescriptorUtils::set_port_descriptor(in, subtensor);
+    }
+    auto subtensor = root_subtensor;
+    if (equation->output(0).get_partial_shape().size() < 2) {
+        subtensor = {root_subtensor.back()};
     }
     ov::snippets::lowered::PortDescriptorUtils::set_port_descriptor(equation->output(0), root_subtensor);
     return true;