Support torch 2.3 and remove version ceiling restriction #5417
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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: | |
concurrency: | |
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }} | |
cancel-in-progress: true | |
env: | |
SPARSEZOO_TEST_MODE: true | |
NM_DISABLE_ANALYTICS: "true" | |
jobs: | |
base-tests: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: [3.8, 3.9, '3.11'] | |
os: [ubuntu-20.04] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- name: Setup Python ${{ matrix.python-version }} | |
uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- 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: pip install sparsezoo/ | |
- name: "Clean sparsezoo directory" | |
run: rm -r sparsezoo/ | |
- name: ⚙️ Install dependencies | |
run: pip install .[dev,server,image_classification,yolov8,transformers,clip] | |
- name: Run base tests | |
run: make test | |
cli-smoke-tests: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: [3.8, 3.9, '3.11'] | |
os: [ubuntu-20.04] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- name: Setup Python ${{ matrix.python-version }} | |
uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- 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: pip install sparsezoo/ | |
- name: "Clean sparsezoo directory" | |
run: rm -r sparsezoo/ | |
- name: ⚙️ Install dependencies | |
run: pip install .[dev,server,image_classification,transformers] | |
- name: Run CLI smoke tests | |
run: PYTEST_ARGS="-m smoke" make test TARGETS=cli,nobase | |
integrations-tests: | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: [3.8, 3.9, '3.10'] | |
os: [ubuntu-20.04] | |
runs-on: ${{ matrix.os }} | |
steps: | |
- name: Setup Python ${{ matrix.python-version }} | |
uses: actions/setup-python@v2 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- 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: pip install sparsezoo/ | |
- name: "Clean sparsezoo directory" | |
run: rm -r sparsezoo/ | |
- name: ⚙️ Install dependencies | |
run: pip install .[dev] | |
- name: Run integrations tests | |
run: make test_integrations |