reshape

chenhu-wang · Jan 3, 2024 · be9bc14 · be9bc14
1 parent d24e71d
commit be9bc14
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/src/common/snippets/include/snippets/generator.hpp b/src/common/snippets/include/snippets/generator.hpp
@@ -104,8 +104,9 @@ class Generator {
     *        gpr->gpr: (Parameter, Result, LoopBegin, LoopEnd etc)
     *        gpr->vec: or vec->gpr Load/LoadConvert, Store/StoreConvert, BroadcastLoad etc.
     *        vec->vec: all other "normal" operations that perform calculations on vector registers: Add, BroadcastMove, Power, etc.
+    *        none->none: some operations that actually will not emit code, no registers is need: Reshape, etc.
     */
-    enum opRegType {gpr2gpr, gpr2vec, vec2gpr, vec2vec};
+    enum opRegType {gpr2gpr, gpr2vec, vec2gpr, vec2vec, none2none};
     /**
      * @brief gets register type by op type
      *        TODO: Should be static attribute of emitters

diff --git a/src/common/snippets/src/generator.cpp b/src/common/snippets/src/generator.cpp
@@ -106,6 +106,8 @@ Generator::opRegType Generator::get_op_reg_type(const std::shared_ptr<Node>& op)
              std::dynamic_pointer_cast<op::HorizonSum>(op) ||
              std::dynamic_pointer_cast<op::Fill>(op))
         return vec2vec;
+    else if (std::dynamic_pointer_cast<ov::op::v1::Reshape>(op))
+        return none2none;
     else
         return get_specific_op_reg_type(op);
 }

diff --git a/src/common/snippets/src/lowered/pass/assign_registers.cpp b/src/common/snippets/src/lowered/pass/assign_registers.cpp
@@ -30,6 +30,8 @@ bool AssignRegisters::run(LinearIR& linear_ir) {
     for (auto& expr : expressions) {
         auto op = expr->get_node();
         auto reg_type = m_reg_type_mapper(op);
+        if (reg_type == Generator::opRegType::none2none)
+            continue;
         typed_ops.emplace_back(reg_type, expr);
         num_parameters += is_type<ov::op::v0::Parameter>(op);
         num_results += is_type<ov::op::v0::Result>(op);
@@ -134,6 +136,8 @@ bool AssignRegisters::run(LinearIR& linear_ir) {
             case Generator::opRegType::vec2gpr:
                 enumerate_out_tensors(t_op.second, regs_gpr, manually_assigned_gprs, counter_gpr);
                 break;
+            default:
+                break;
         }
     }
     // todo: make one for gpr and one for vector
@@ -177,6 +181,8 @@ bool AssignRegisters::run(LinearIR& linear_ir) {
                 used_vec[i] = tensor2reg(used_tensors, regs_vec);
                 defined_gpr[i] = tensor2reg(defined_tensors, regs_gpr);
                 break;
+            default:
+                break;
         }
     }
 
@@ -225,6 +231,8 @@ bool AssignRegisters::run(LinearIR& linear_ir) {
                         case Generator::opRegType::gpr2vec:
                             life_out_gpr[n].insert(life_in_gpr[k].begin(), life_in_gpr[k].end());
                             break;
+                        default:
+                            break;
                     }
                 }
             }