Update test-models.yml

This commit is contained in:
powderluv
2022-06-30 13:38:31 -07:00
committed by GitHub
parent b012b9a5f7
commit cf4d363436

View File

@@ -1,7 +1,7 @@
# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions
name: Validate torch-models on Shark Runtime
name: Validate Models on Shark Runtime
on:
push:
@@ -53,13 +53,13 @@ jobs:
${{ matrix.os }}-pip-
- name: Install dependencies
if: matrix.suite == lint
if: matrix.suite == 'lint'
run: |
python -m pip install --upgrade pip
python -m pip install flake8 pytest toml black
- name: Lint with flake8
if: matrix.suite == lint
if: matrix.suite == 'lint'
run: |
# black format check
black --version
@@ -70,7 +70,7 @@ jobs:
flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics --exclude lit.cfg.py
- name: Validate CPU Models
if: matrix.suite == cpu
if: matrix.suite == 'cpu'
run: |
cd $GITHUB_WORKSPACE
PYTHON=python${{ matrix.python-version }} IMPORTER=1 ./setup_venv.sh
@@ -78,7 +78,7 @@ jobs:
pytest -k 'not benchmark' --ignore=tank/tf/ --ignore=benchmarks/tests/test_hf_benchmark.py --ignore=benchmarks/tests/test_benchmark.py
- name: Validate GPU Models
if: matrix.suite == gpu
if: matrix.suite == 'gpu'
run: |
cd $GITHUB_WORKSPACE
PYTHON=python${{ matrix.python-version }} IMPORTER=1 ./setup_venv.sh
@@ -86,7 +86,7 @@ jobs:
pytest -k 'gpu' --ignore=shark/tests/test_shark_importer.py --ignore=benchmarks/tests/test_hf_benchmark.py --ignore=benchmarks/tests/test_benchmark.py
- name: Validate Vulkan Models
if: matrix.suite == vulkan
if: matrix.suite == 'vulkan'
run: |
cd $GITHUB_WORKSPACE
PYTHON=python${{ matrix.python-version }} IMPORTER=1 ./setup_venv.sh