Split perf-builder into cpu/gpu/vulkan

This commit is contained in:
powderluv
2022-06-16 14:08:36 -07:00
committed by GitHub
parent c8230fe093
commit 918b36c177

View File

@@ -70,7 +70,7 @@ jobs:
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:
perf-linux-cpu:
runs-on: a100
timeout-minutes: 300
continue-on-error: true
@@ -99,4 +99,68 @@ jobs:
cd $GITHUB_WORKSPACE
IMPORTER=1 ./setup_venv.sh
source shark.venv/bin/activate
pytest --ignore=shark/tests/test_shark_importer.py
pytest -k 'cpu' --ignore=shark/tests/test_shark_importer.py
perf-linux-cuda:
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 'gpu' --ignore=shark/tests/test_shark_importer.py
perf-linux-vulkan:
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 'vulkan' --ignore=shark/tests/test_shark_importer.py