github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-10 06:17:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Modify pytest to run a certain test.

Browse Source
This commit is contained in:
Prashant Kumar
2022-05-17 11:13:51 +00:00
parent 3eb222fbdd
commit 63f9b206ac
2 changed files with 9 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pytest.ini
View File

@@ -1,4 +1,4 @@
[pytest]
addopts = --workers auto --verbose -p no:warnings
addopts = --workers auto -m cpu_static --verbose -p no:warnings
norecursedirs = inference

30
shark/examples/test_models.py
View File

@@ -78,29 +78,15 @@ def compare_tensors(torch_tensor, numpy_tensor):
############################# Model Tests ####################################
# Please remove the skip flag to run the test with required param.
# A specific case can be run with -m parameter. For eg., to run gpu - static
# case for all models it can be run by `pytest -m gpu_static` command.
pytest_param = pytest.mark.parametrize(('dynamic', 'device'), [
pytest.param(False, 'cpu'),
pytest.param(
True,
'cpu',
marks=pytest.mark.skip(reason="dynamic-shape is not supported.")),
pytest.param(
False,
'gpu',
marks=pytest.mark.skip(reason="device in the CI not supported.")),
pytest.param(
True,
'gpu',
marks=pytest.mark.skip(reason="device in the CI not supported.")),
pytest.param(
False,
'vulkan',
marks=pytest.mark.skip(reason="device in the CI not supported.")),
pytest.param(
True,
'vulkan',
marks=pytest.mark.skip(reason="device in the CI not supported.")),
pytest.param(False, 'cpu', marks=pytest.mark.cpu_static),
pytest.param(True, 'cpu', marks=pytest.mark.cpu_dynamic),
pytest.param(False, 'gpu', marks=pytest.mark.gpu_static),
pytest.param(True, 'gpu', marks=pytest.mark.gpu_dynamic),
pytest.param(False, 'vulkan', marks=pytest.mark.vulkan_static),
pytest.param(True, 'vulkan', marks=pytest.mark.vulkan_dynamic),
])