Fixed InsertStore for Buffer wo inputs

openvinotoolkit · May 17, 2023 · ec5920b · ec5920b
1 parent f71b552
commit ec5920b
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/common/snippets/src/lowered/pass/insert_load_store.cpp b/src/common/snippets/src/lowered/pass/insert_load_store.cpp
@@ -140,12 +140,19 @@ bool InsertLoadStore::run(LinearIR& linear_ir) {
     for (auto expr_it = linear_ir.begin(); expr_it != linear_ir.end(); expr_it++) {
         const auto expr = *expr_it;
         const auto& node = expr->get_node();
-        if (ov::is_type<opset1::Parameter>(node) || ov::is_type<op::Buffer>(node)) {
+        if (ov::is_type<opset1::Parameter>(node)) {
             modified |= insert_load(linear_ir, expr_it);
+            continue;
         }
-
-        if (ov::is_type<opset1::Result>(node) || ov::is_type<op::Buffer>(node)) {
+        if (ov::is_type<opset1::Result>(node)) {
             modified |= insert_store(linear_ir, expr_it);
+            continue;
+        }
+        if (auto buffer = ov::as_type_ptr<op::Buffer>(node)) {
+            modified |= insert_load(linear_ir, expr_it);
+            if (buffer->is_intermediate_memory())
+                modified |= insert_store(linear_ir, expr_it);
+            continue;
         }
     }