From 68ef1555bcae1ef80ca120745a61e2349c5baee0 Mon Sep 17 00:00:00 2001 From: Mateusz Tabaka Date: Thu, 5 May 2022 09:55:02 +0200 Subject: [PATCH] Enable test_loop_simple_precommit on GPU (#11625) --- tests/layer_tests/onnx_tests/test_loop.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/layer_tests/onnx_tests/test_loop.py b/tests/layer_tests/onnx_tests/test_loop.py index 1a315a37a5c..919b2e069db 100644 --- a/tests/layer_tests/onnx_tests/test_loop.py +++ b/tests/layer_tests/onnx_tests/test_loop.py @@ -274,8 +274,6 @@ class TestLoop(OnnxRuntimeLayerTest): @pytest.mark.precommit @pytest.mark.timeout(250) def test_loop_simple_precommit(self, ie_device, precision, ir_version, temp_dir, api_2): - if ie_device == 'GPU': - pytest.skip('Loop not supported on GPU') self._test(*self.create_loop(), ie_device, precision, ir_version, temp_dir=temp_dir, infer_timeout=150, api_2=api_2) @@ -283,6 +281,6 @@ class TestLoop(OnnxRuntimeLayerTest): @pytest.mark.timeout(250) def test_loop_in_loop_simple_precommit(self, ie_device, precision, ir_version, temp_dir, api_2): if ie_device == 'GPU': - pytest.skip('Loop not supported on GPU') + pytest.xfail("Program doesn't contain primitive: constant:res/10/M_2 that is input to: loop") self._test(*self.create_loop_in_loop(), ie_device, precision, ir_version, temp_dir=temp_dir, infer_timeout=150, api_2=api_2)