Skip to content

[Text Generation] Automatically benchmark in auto-regressive setting #3279

[Text Generation] Automatically benchmark in auto-regressive setting

[Text Generation] Automatically benchmark in auto-regressive setting #3279

Workflow file for this run

name: Test Checks
on:
push:
branches:
- main
- 'release/*'
pull_request:
branches:
- main
- 'release/*'
# Allows you to run this workflow manually from the Actions tab
workflow_dispatch:
env:
SPARSEZOO_TEST_MODE: true
NM_DISABLE_ANALYTICS: "true"
jobs:
base-tests:
runs-on: ubuntu-20.04
steps:
- uses: actions/checkout@v2
- uses: actions/checkout@v2
with:
repository: "neuralmagic/sparsezoo"
path: "sparsezoo"
ref: ${{needs.test-setup.outputs.branch}}
- name: "⚙️ Install sparsezoo dependencies"
run: pip3 install -U pip && pip3 install setuptools sparsezoo/
- name: "Clean sparsezoo directory"
run: rm -r sparsezoo/
- name: ⚙️ Install dependencies
run: pip3 install .[dev,server,image_classification,transformers] opencv-python
- name: Run base tests
run: make test
cli-smoke-tests:
runs-on: ubuntu-20.04
steps:
- uses: actions/checkout@v2
- uses: actions/checkout@v2
with:
repository: "neuralmagic/sparsezoo"
path: "sparsezoo"
ref: ${{needs.test-setup.outputs.branch}}
- name: "⚙️ Install sparsezoo dependencies"
run: pip3 install -U pip && pip3 install setuptools sparsezoo/
- name: "Clean sparsezoo directory"
run: rm -r sparsezoo/
- name: ⚙️ Install dependencies
run: pip3 install .[dev,server,image_classification,transformers]
- name: Run CLI smoke tests
run: PYTEST_ARGS="-m smoke" make test TARGETS=cli,nobase
integrations-tests:
runs-on: ubuntu-20.04
steps:
- uses: actions/checkout@v2
- uses: actions/checkout@v2
with:
repository: "neuralmagic/sparsezoo"
path: "sparsezoo"
ref: ${{needs.test-setup.outputs.branch}}
- name: "⚙️ Install sparsezoo dependencies"
run: pip3 install -U pip && pip3 install setuptools sparsezoo/
- name: "Clean sparsezoo directory"
run: rm -r sparsezoo/
- name: ⚙️ Install dependencies
run: pip3 install .[dev,server,image_classification,transformers,haystack]
- name: Run integrations tests
run: make test_integrations