IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fall back to legacy path if input freezing feature is used (#9668)

Browse Source
This commit is contained in:
Mateusz Bencer
2022-01-21 14:46:59 +01:00
committed by GitHub
parent 8ebe75f81a
commit 82cc7165e7
4 changed files with 52 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
tools/mo/unit_tests/mo/utils/mo_fallback_test_actual.py
View File

@@ -404,3 +404,45 @@ class TestMoFallback(unittest.TestCase):
with pytest.raises(Error) as ex: # not called
prepare_ir(args)
assert str(ex) == 'Legacy transformations configuration is not supported for the new frontend'
@generate(*[('in1->[5.0 5.0 7.0 8.0],in2->[1.0 2.0 3.0 4.0]', None, None, 'mo_legacy', 'input_freezing'), #fallback
('in1,in2->[1.0 2.0 3.0 4.0]', None, None, 'mo_legacy', 'input_freezing'), #fallback
('in1->[1.0 2.0 3.0 4.0],in2', True, None, 'onnx_frontend', None),
('in1->[1.0 2.0 3.0 4.0],in2', None, True, 'mo_legacy', None),
])
def test_fallback_if_input_freezing_used(self, input, use_new_fe, use_legacy, conversion_method, fallback_reason):
with patch('openvino.tools.mo.main.get_default_frontends') as default_fe:
default_fe.return_value = get_test_default_frontends()
args = base_args_config(use_legacy, use_new_fe)
args.input_model = "test_model.onnx"
args.input = input
prepare_ir(args)
tm.Telemetry.send_event.assert_any_call('mo', 'conversion_method', conversion_method)
if fallback_reason:
tm.Telemetry.send_event.assert_any_call('mo', 'fallback_reason', fallback_reason)
else:
with pytest.raises(AssertionError): # not called
tm.Telemetry.send_event.assert_any_call('mo', 'fallback_reason', fallback_reason)
@generate(*[('in2->[1.0 2.0 3.0 4.0]', None, None, 'mo_legacy', 'input_freezing'), #fallback
('in2->[1.0 2.0 3.0 4.0]', True, None, 'onnx_frontend', None),
('in2->[1.0 2.0 3.0 4.0]', None, True, 'mo_legacy', None),
])
def test_fallback_if_input_freezing_used_in_deprecated_way(self, input_freezing_value, use_new_fe, use_legacy, conversion_method, fallback_reason):
with patch('openvino.tools.mo.main.get_default_frontends') as default_fe:
default_fe.return_value = get_test_default_frontends()
args = base_args_config(use_legacy, use_new_fe)
args.input_model = "test_model.onnx"
args.freeze_placeholder_with_value = input_freezing_value
prepare_ir(args)
tm.Telemetry.send_event.assert_any_call('mo', 'conversion_method', conversion_method)
if fallback_reason:
tm.Telemetry.send_event.assert_any_call('mo', 'fallback_reason', fallback_reason)
else:
with pytest.raises(AssertionError): # not called
tm.Telemetry.send_event.assert_any_call('mo', 'fallback_reason', fallback_reason)