diff --git a/src/common/transformations/src/transformations/mlir/convert_common.cpp b/src/common/transformations/src/transformations/mlir/convert_common.cpp index 45ab7a904f0d90..dee142a554fe79 100644 --- 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 index 1459ae34ef4c07..6eb56846145510 100644 --- 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(); + const auto& rti = node->get_rt_info(); + if(!rti.count("__subgraph_id")) { + return nullptr; + } + auto id = rti.at("__subgraph_id").as(); if(id) { id = ov::symbol::ancestor_of(id); }