Update test-models.yml

disable a100 cpu for now
This commit is contained in:
powderluv
2022-06-16 16:27:11 -07:00
committed by GitHub
parent ce7350f5f8
commit 0093610054

View File

@@ -69,39 +69,7 @@ jobs:
PYTHON=python3.10 IMPORTER=1 ./setup_venv.sh
source shark.venv/bin/activate
pytest -k 'not benchmark' --ignore=benchmarks/tests/test_hf_benchmark.py --ignore=benchmarks/tests/test_benchmark.py --ignore=tank/tf/ --ignore=shark/tests/test_shark_importer.py
perf-linux-cpu:
runs-on: a100
timeout-minutes: 300
continue-on-error: true
strategy:
fail-fast: false
matrix:
python-version: ["3.10"]
steps:
- uses: actions/checkout@v3
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v3
with:
python-version: ${{ matrix.python-version }}
- name: Setup pip cache
uses: actions/cache@v3
with:
path: ~/.cache/pip
key: ${{ runner.os }}-pip-${{ hashFiles('**/requirements.txt') }}
restore-keys: |
${{ runner.os }}-pip-
- name: Validate Models
run: |
cd $GITHUB_WORKSPACE
IMPORTER=1 ./setup_venv.sh
source shark.venv/bin/activate
pytest -k 'cpu' --ignore=shark/tests/test_shark_importer.py
perf-linux-cuda:
runs-on: a100
timeout-minutes: 300