From 030da611e621eefdc2fa25487f7b9cd1f4dc60a5 Mon Sep 17 00:00:00 2001 From: Tomasz Jankowski Date: Mon, 30 Sep 2024 12:47:45 +0200 Subject: [PATCH] MO Patch reading custom rt info --- .../tools/mo/utils/ir_engine/ir_engine.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py b/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py index aec4049c6b32e8..e58215d7e1c192 100644 --- a/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py +++ b/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py @@ -539,10 +539,14 @@ def __read_old_api_map_element_type(attr, layer_type): @staticmethod def __read_rt_info_common(attr): - attr_name = attr.attrib['name'] - version = int(attr.attrib['version']) - rt_info = OrderedDict() - for key in attr.attrib: - if key not in ('name', 'version'): - rt_info[key] = attr.attrib[key] - return {(attr_name, version): rt_info} + # ignore custom info, MO is deprecated so there's no needed for forward compatibility + if 'version' in attr.attrib: + attr_name = attr.attrib['name'] + version = int(attr.attrib['version']) + rt_info = OrderedDict() + for key in attr.attrib: + if key not in ('name', 'version'): + rt_info[key] = attr.attrib[key] + return {(attr_name, version): rt_info} + else: + return {}