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

Add conditions for CPU/GPU failures

Browse Source
This commit is contained in:
Ilina 2023-03-13 17:38:52 +01:00
parent face0bb331
commit 790524c59c
5 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/layer_tests/onnx_tests/test_lrn.py
View File

@ -120,9 +120,10 @@ class TestLRN(OnnxRuntimeLayerTest):
@pytest.mark.parametrize("params", test_data)
@pytest.mark.nightly
@pytest.mark.xfail(reason='Mark as xfailed all failed onnx layer tests')
def test_lrn(self, params, ie_device, precision, ir_version, temp_dir, use_old_api):
# onnxruntime only supports 4D tensors for LRN
if ie_device == 'GPU':
pytest.xfail('Mark as xfailed all failed onnx layer tests')
self.skip_framework = True
self._test(*self.create_net(**params, ir_version=ir_version), ie_device, precision,
ir_version,

6
tests/layer_tests/onnx_tests/test_matmul.py
View File

@ -173,8 +173,9 @@ class TestMatMul(OnnxRuntimeLayerTest):
@pytest.mark.parametrize("params", test_data_broadcasting)
@pytest.mark.nightly
@pytest.mark.xfail(reason='Mark as xfailed all failed onnx layer tests')
def test_matmul_bc(self, params, ie_device, precision, ir_version, temp_dir, use_old_api):
if ie_device == 'GPU':
pytest.xfail('Mark as xfailed all failed onnx layer tests')
self._test(*self.create_net(**params, precision=precision, ir_version=ir_version),
ie_device, precision, ir_version, temp_dir=temp_dir, use_old_api=use_old_api)
@ -187,8 +188,9 @@ class TestMatMul(OnnxRuntimeLayerTest):
@pytest.mark.parametrize("params", test_data_broadcasting)
@pytest.mark.nightly
@pytest.mark.xfail(reason='Mark as xfailed all failed onnx layer tests')
def test_dual_matmul_bc(self, params, ie_device, precision, ir_version, temp_dir, use_old_api):
if ie_device == 'GPU':
pytest.xfail('Mark as xfailed all failed onnx layer tests')
self._test(*self.create_dual_net(**params, ir_version=ir_version), ie_device, precision,
ir_version,
temp_dir=temp_dir, use_old_api=use_old_api)

3
tests/layer_tests/onnx_tests/test_reduce_lp.py
View File

@ -230,9 +230,10 @@ class TestReduceL1L2(OnnxRuntimeLayerTest):
@pytest.mark.parametrize("keep_dims", [True, False])
@pytest.mark.parametrize("reduce_p", [1, 2])
@pytest.mark.nightly
@pytest.mark.xfail(reason='Mark as xfailed all failed onnx layer tests')
def test_reduce_lp(self, params, keep_dims, reduce_p, ie_device, precision, ir_version,
temp_dir, use_old_api):
if ie_device == 'GPU':
pytest.xfail('Mark as xfailed all failed onnx layer tests')
self._test(*self.create_reduce_lp(**params, keep_dims=keep_dims, reduce_p=reduce_p,
ir_version=ir_version),
ie_device, precision, ir_version, temp_dir=temp_dir, use_old_api=use_old_api)

6
tests/layer_tests/onnx_tests/test_slice.py
View File

@ -395,15 +395,17 @@ class TestSlice(OnnxRuntimeLayerTest):
@pytest.mark.parametrize("params", test_data_no_steps + test_data_with_steps)
@pytest.mark.nightly
@pytest.mark.xfail(reason='Mark as xfailed all failed onnx layer tests')
def test_slice_const_opset10(self, params, ie_device, precision, ir_version, temp_dir, use_old_api):
if ie_device == 'GPU':
pytest.xfail('Mark as xfailed all failed onnx layer tests')
self._test(*self.create_net_const(**params, opset=10, ir_version=ir_version),
ie_device, precision, ir_version, temp_dir=temp_dir, use_old_api=use_old_api)
@pytest.mark.parametrize("params", test_data_no_steps + test_data_with_steps)
@pytest.mark.nightly
@pytest.mark.xfail(reason='Mark as xfailed all failed onnx layer tests')
def test_slice_opset11(self, params, ie_device, precision, ir_version, temp_dir, use_old_api):
if ie_device == 'GPU':
pytest.xfail('Mark as xfailed all failed onnx layer tests')
self._test(
*self.create_net(**params, opset=11, ir_version=ir_version), ie_device, precision,
ir_version,

3
tests/layer_tests/onnx_tests/test_topk.py
View File

@ -155,8 +155,9 @@ class TestTopK(OnnxRuntimeLayerTest):
@pytest.mark.parametrize("params", test_data)
@pytest.mark.nightly
@pytest.mark.xfail(reason='Mark as xfailed all failed onnx layer tests')
def test_topk_opset10(self, params, ie_device, precision, ir_version, temp_dir, use_old_api):
if ie_device == 'CPU':
pytest.xfail('Mark as xfailed all failed onnx layer tests')
self._test(*self.create_net(**params, opset=10, ir_version=ir_version), ie_device,
precision, ir_version,
temp_dir=temp_dir, use_old_api=use_old_api)