Ensure CPU workflow job suite only collects CPU tests.

This commit is contained in:
monorimet
2022-07-16 13:50:36 -05:00
parent c5ae01311d
commit 5da32601c0

View File

@@ -77,7 +77,7 @@ jobs:
cd $GITHUB_WORKSPACE
PYTHON=python${{ matrix.python-version }} IMPORTER=1 ./setup_venv.sh
source shark.venv/bin/activate
pytest -n 2 -k 'not benchmark' --ignore=shark/tests/test_shark_importer.py --ignore=tank/tf/ --ignore=benchmarks/tests/test_hf_benchmark.py --ignore=benchmarks/tests/test_benchmark.py
pytest -n 2 -k 'cpu' --ignore=shark/tests/test_shark_importer.py --ignore=benchmarks/tests/test_hf_benchmark.py --ignore=benchmarks/tests/test_benchmark.py
- name: Validate GPU Models
if: matrix.suite == 'gpu'