feat: add onnxslim support #1903
This file contains hidden or 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
| # NOTE: Make sure this file is consistent with .gitlab/tests.yml | |
| name: Unit tests | |
| on: | |
| pull_request: | |
| branches: [main, release/*, feature/*] | |
| push: | |
| branches: [main, release/*, feature/*] | |
| paths: | |
| - ".github/workflows/unit_tests.yml" | |
| - "modelopt/**" | |
| - "tests/unit/**" | |
| - "setup.py" | |
| - "tox.ini" | |
| schedule: | |
| - cron: "0 0 * * *" # Nightly | |
| workflow_dispatch: # On-demand | |
| # Cancel previous runs if new commit is pushed | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }} | |
| cancel-in-progress: true | |
| jobs: | |
| check-dco: | |
| uses: ./.github/workflows/_wait_for_checks.yml | |
| permissions: | |
| checks: read | |
| secrets: inherit | |
| with: | |
| match_pattern: "^DCO$" | |
| linux: | |
| needs: [check-dco] | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 30 | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-python@v5 | |
| with: | |
| python-version: "3.12" | |
| - name: Run unit tests | |
| run: pip install tox && COV_ARGS="--cov" tox -e py312-torch29-tf_latest-unit | |
| - name: Upload coverage reports to Codecov | |
| uses: codecov/codecov-action@v5 | |
| with: | |
| token: ${{ secrets.CODECOV_TOKEN }} | |
| fail_ci_if_error: true | |
| verbose: true | |
| windows: | |
| if: github.event_name == 'pull_request' | |
| needs: [linux] | |
| runs-on: windows-latest | |
| timeout-minutes: 30 | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-python@v5 | |
| with: | |
| python-version: "3.12" | |
| - name: Run unit tests (without coverage) | |
| run: pip install tox && tox -e py312-torch29-tf_latest-unit | |
| multi-py: | |
| if: github.event_name == 'pull_request' | |
| needs: [linux] | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 30 | |
| strategy: | |
| matrix: | |
| py: [10, 11] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-python@v5 | |
| with: | |
| python-version: "3.${{ matrix.py }}" | |
| - name: Run unit tests | |
| run: pip install tox && tox -e py3${{ matrix.py }}-torch29-tf_latest-unit | |
| multi-torch: | |
| if: github.event_name == 'pull_request' | |
| needs: [linux] | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 30 | |
| strategy: | |
| matrix: | |
| torch: [26, 27, 28] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-python@v5 | |
| with: | |
| python-version: "3.12" | |
| - name: Run unit tests | |
| run: pip install tox && tox -e py312-torch${{ matrix.torch }}-tf_latest-unit | |
| multi-transformers: | |
| if: github.event_name == 'pull_request' | |
| needs: [linux] | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 30 | |
| strategy: | |
| matrix: | |
| tf: [min] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-python@v5 | |
| with: | |
| python-version: "3.12" | |
| - name: Run unit tests | |
| run: pip install tox && tox -e py312-torch29-tf_${{ matrix.tf }}-unit | |
| partial-install: | |
| if: github.event_name == 'pull_request' | |
| needs: [linux] | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 30 | |
| strategy: | |
| matrix: | |
| test-env: [onnx, torch] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-python@v5 | |
| with: | |
| python-version: "3.12" | |
| - name: Run unit tests | |
| run: pip install tox && tox -e py312-partial-unit-${{ matrix.test-env }} | |
| unit-pr-required-check: | |
| # Run even if some jobs are skipped | |
| if: ${{ github.event_name == 'pull_request' && always() }} | |
| needs: [linux, windows, multi-py, multi-torch, multi-transformers, partial-install] | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Required unit tests did not succeed | |
| if: >- | |
| ${{ needs.linux.result != 'success' || | |
| needs.windows.result != 'success' || | |
| needs.multi-py.result != 'success' || | |
| needs.multi-torch.result != 'success' || | |
| needs.multi-transformers.result != 'success' || | |
| needs.partial-install.result != 'success' }} | |
| run: exit 1 |