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

Enable test_loop_simple_precommit on GPU (#11625)

Browse Source
This commit is contained in:
Mateusz Tabaka 2022-05-05 09:55:02 +02:00 committed by GitHub
parent 0babf20bd2
commit 68ef1555bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/layer_tests/onnx_tests/test_loop.py
View File

@ -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)