Fix graph partitioner when two MLIROp instances go side-by-side. Fix …

…in return statement in is_dubug function. (#157)
slyalin · Jul 30, 2024 · 2a646fb · 2a646fb
1 parent 09a1342
commit 2a646fb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/common/transformations/src/transformations/mlir/convert_common.cpp b/src/common/transformations/src/transformations/mlir/convert_common.cpp
@@ -62,7 +62,7 @@ namespace ov {
 namespace mlir {
 
 bool is_debug() {
-    util::getenv_bool("OV_MLIR_DEBUG", false);
+    return util::getenv_bool("OV_MLIR_DEBUG", false);
 }
 
 Location createLayerLocation(MLIRContext* ctx, const std::string& layerName, const std::string& layerType) {

diff --git a/src/common/transformations/src/transformations/mlir/subgraph_tracker.cpp b/src/common/transformations/src/transformations/mlir/subgraph_tracker.cpp
@@ -116,7 +116,11 @@ void SubgraphTracker::set_dependencies(NodePtr node, const Dependencies& depende
 // set/get subgraph id that a give node belongs to
 
 SubgraphID SubgraphTracker::get_subgraph_id(NodePtr node) {
-    auto id = node->get_rt_info().at("__subgraph_id").as<SubgraphID>();
+    const auto& rti = node->get_rt_info();
+    if(!rti.count("__subgraph_id")) {
+        return nullptr;
+    }
+    auto id = rti.at("__subgraph_id").as<SubgraphID>();
     if(id) {
         id = ov::symbol::ancestor_of(id);
     }