Vladislav comments apply 2

chenhu-wang · Jun 12, 2024 · 35195e7 · 35195e7
1 parent 87cb1e8
commit 35195e7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/src/common/snippets/include/snippets/LIR_pass_dumper.hpp b/src/common/snippets/include/snippets/LIR_pass_dumper.hpp
@@ -35,14 +35,18 @@ class LIRPassDump {
             std::string xml_path = pathAndName + std::to_string(num) + '_' + pass_name + "_control_flow" + postfix +".xml";
             lowered::pass::SerializeControlFlow SerializeLIR(xml_path);
             SerializeLIR.run(linear_ir);
-            num++;
         }
-        if (debug_config.dumpLIR.format.filter[DebugCapsConfig::LIRFormatFilter::dataFlow]) {
-            std::string xml_path = pathAndName + std::to_string(num) + '_' + pass_name + "_data_flow" + postfix +".xml";
-            lowered::pass::SerializeDataFlow SerializeLIR(xml_path);
-            SerializeLIR.run(linear_ir);
-            num++;
+        try {
+            if (debug_config.dumpLIR.format.filter[DebugCapsConfig::LIRFormatFilter::dataFlow]) {
+                std::string xml_path = pathAndName + std::to_string(num) + '_' + pass_name + "_data_flow" + postfix +".xml";
+                lowered::pass::SerializeDataFlow SerializeLIR(xml_path);
+                SerializeLIR.run(linear_ir);
+            }
+        }
+        catch (const std::exception& exp) {
+            printf("[WARNING] Dump LIR on pass %s failed. Data flow can't be dumped on LIR after InsertSpecificIteration was applied.", pass_name.c_str());
         }
+        num++;
     }
 
     lowered::LinearIR& linear_ir;

diff --git a/src/common/snippets/include/snippets/lowered/linear_ir.hpp b/src/common/snippets/include/snippets/lowered/linear_ir.hpp
@@ -9,10 +9,7 @@
 #include "snippets/lowered/expression.hpp"
 #include "snippets/target_machine.hpp"
 #include "snippets/shape_inference/shape_inference.hpp"
-
-#ifdef SNIPPETS_DEBUG_CAPS
 #include "snippets/debug_caps.hpp"
-#endif
 
 namespace ov {
 namespace snippets {