Refactor Quantization Modifer and Reloading #8569
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/*' | |
| jobs: | |
| test-setup: | |
| runs-on: ubuntu-22.04 | |
| outputs: | |
| branch: ${{ steps.get-branch.outputs.branch }} | |
| base: ${{ steps.base-check.outputs.output }} | |
| deepsparse: ${{ steps.deepsparse-check.outputs.output }} | |
| onnx: ${{ steps.onnx-check.outputs.output }} | |
| pytorch: ${{ steps.pytorch-check.outputs.output }} | |
| export: ${{ steps.export-check.outputs.output }} | |
| transformers: ${{ steps.transformers-check.outputs.output }} | |
| steps: | |
| - uses: actions/checkout@v2 | |
| with: | |
| fetch-depth: 0 | |
| - run: git branch --show-current | |
| - name: Get current branch | |
| id: get-branch | |
| run: > | |
| (git branch --show-current | grep -E "release/") | |
| && echo "::set-output name=branch::$(git branch --show-current)" | |
| || echo "::set-output name=branch::main" | |
| - name: "Checking if sparseml was changed" | |
| id: base-check | |
| run: > | |
| ((git diff --name-only origin/main HEAD | grep -E "[src|tests]/sparseml|setup.py|.github") | |
| || (echo $GITHUB_REF | grep -E "refs/heads/[release/|main]")) | |
| && echo "::set-output name=output::1" || echo "::set-output name=output::0" | |
| - name: "Checking if sparseml.deepsparse was changed" | |
| id: deepsparse-check | |
| run: > | |
| ((git diff --name-only origin/main HEAD | grep -E "[src|tests]/sparseml/deepsparse|setup.py|.github") | |
| || (echo $GITHUB_REF | grep -E "refs/heads/[release/|main]")) | |
| && echo "::set-output name=output::1" || echo "::set-output name=output::0" | |
| - name: "Checking if sparseml.onnx was changed" | |
| id: onnx-check | |
| run: > | |
| ((git diff --name-only origin/main HEAD | grep -E "[src|tests]/sparseml/onnx|setup.py|.github") | |
| || (echo $GITHUB_REF | grep -E "refs/heads/[release/|main]")) | |
| && echo "::set-output name=output::1" || echo "::set-output name=output::0" | |
| - name: "Checking if sparseml.pytorch was changed" | |
| id: pytorch-check | |
| run: > | |
| ((git diff --name-only origin/main HEAD | grep -E "[src|tests]/sparseml/modifiers|pytorch|setup.py|.github") | |
| || (echo $GITHUB_REF | grep -E "refs/heads/[release/|main]")) | |
| && echo "::set-output name=output::1" || echo "::set-output name=output::0" | |
| - name: "Checking if sparseml.export was changed" | |
| id: export-check | |
| run: > | |
| ((git diff --name-only origin/main HEAD | grep -E "[src|tests]/sparseml/export|setup.py|.github") | |
| || (echo $GITHUB_REF | grep -E "refs/heads/[release/|main]")) | |
| && echo "::set-output name=output::1" || echo "::set-output name=output::0" | |
| - name: "Checking if sparseml.transformers was changed" | |
| id: transformers-check | |
| run: > | |
| ((git diff --name-only origin/main HEAD | grep -E "[src|tests]/sparseml/transformers|setup.py|.github") | |
| || (echo $GITHUB_REF | grep -E "refs/heads/[release/|main]")) | |
| && echo "::set-output name=output::1" || echo "::set-output name=output::0" | |
| base-tests: | |
| runs-on: ubuntu-22.04 | |
| env: | |
| SPARSEZOO_TEST_MODE: "true" | |
| needs: test-setup | |
| if: ${{needs.test-setup.outputs.base == 1}} | |
| steps: | |
| - uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.11' | |
| - 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,onnxruntime] | |
| - name: "🔬 Running base tests" | |
| run: make test | |
| deepsparse-tests: | |
| runs-on: ubuntu-22.04 | |
| env: | |
| SPARSEZOO_TEST_MODE: "true" | |
| needs: test-setup | |
| if: ${{needs.test-setup.outputs.deepsparse == 1}} | |
| steps: | |
| - uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.11' | |
| - 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,deepsparse,onnxruntime] | |
| - name: "🔬 Running deepsparse tests" | |
| run: make test TARGETS=deepsparse | |
| onnx-tests: | |
| runs-on: ubuntu-22.04 | |
| env: | |
| SPARSEZOO_TEST_MODE: "true" | |
| needs: test-setup | |
| if: ${{needs.test-setup.outputs.onnx == 1}} | |
| steps: | |
| - uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.11' | |
| - 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,torchvision,onnxruntime] | |
| - name: "🔬 Running onnx tests" | |
| run: make test TARGETS=onnx | |
| pytorch-tests: | |
| runs-on: aws-avx2-64G | |
| env: | |
| SPARSEZOO_TEST_MODE: "true" | |
| CLEARML_WEB_HOST: ${{ secrets.CLEARML_WEB_HOST }} | |
| CLEARML_API_HOST: ${{ secrets.CLEARML_API_HOST }} | |
| CLEARML_API_ACCESS_KEY: ${{ secrets.CLEARML_API_ACCESS_KEY }} | |
| CLEARML_FILES_HOST: ${{ secrets.CLEARML_FILES_HOST }} | |
| CLEARML_API_SECRET_KEY: ${{ secrets.CLEARML_API_SECRET_KEY }} | |
| needs: test-setup | |
| if: ${{needs.test-setup.outputs.pytorch == 1}} | |
| steps: | |
| - uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.11' | |
| - 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,torchvision,onnxruntime,transformers] | |
| - name: "🔬 Running pytorch tests" | |
| run: make test TARGETS=pytorch | |
| compat-pytorch-1_9-pytorch-tests: | |
| runs-on: aws-avx2-64G | |
| env: | |
| SPARSEZOO_TEST_MODE: "true" | |
| CLEARML_WEB_HOST: ${{ secrets.CLEARML_WEB_HOST }} | |
| CLEARML_API_HOST: ${{ secrets.CLEARML_API_HOST }} | |
| CLEARML_API_ACCESS_KEY: ${{ secrets.CLEARML_API_ACCESS_KEY }} | |
| CLEARML_FILES_HOST: ${{ secrets.CLEARML_FILES_HOST }} | |
| CLEARML_API_SECRET_KEY: ${{ secrets.CLEARML_API_SECRET_KEY }} | |
| needs: test-setup | |
| if: ${{needs.test-setup.outputs.pytorch == 1}} | |
| steps: | |
| - uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.9' | |
| - 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,torchvision,onnxruntime,transformers] | |
| - name: "🔬 Running pytorch tests" | |
| run: make test TARGETS=pytorch | |
| compat-pytorch-1_9-onnx-tests: | |
| runs-on: ubuntu-22.04 | |
| env: | |
| SPARSEZOO_TEST_MODE: "true" | |
| needs: test-setup | |
| if: ${{needs.test-setup.outputs.onnx == 1}} | |
| steps: | |
| - uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.9' | |
| - 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,torchvision,onnxruntime] torch==1.9.1 | |
| - name: "🔬 Running onnx tests" | |
| run: make test TARGETS=onnx | |
| transformers-tests: | |
| runs-on: aws-avx2-64G | |
| env: | |
| SPARSEZOO_TEST_MODE: "true" | |
| CLEARML_WEB_HOST: ${{ secrets.CLEARML_WEB_HOST }} | |
| CLEARML_API_HOST: ${{ secrets.CLEARML_API_HOST }} | |
| CLEARML_API_ACCESS_KEY: ${{ secrets.CLEARML_API_ACCESS_KEY }} | |
| CLEARML_FILES_HOST: ${{ secrets.CLEARML_FILES_HOST }} | |
| CLEARML_API_SECRET_KEY: ${{ secrets.CLEARML_API_SECRET_KEY }} | |
| needs: test-setup | |
| if: ${{needs.test-setup.outputs.transformers == 1}} | |
| steps: | |
| - uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.11' | |
| - 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,torch,transformers,onnxruntime] | |
| - name: "🔬 Running transformers tests" | |
| run: make test TARGETS=transformers | |
| export-tests: | |
| runs-on: ubuntu-22.04 | |
| env: | |
| SPARSEZOO_TEST_MODE: "true" | |
| needs: test-setup | |
| if: ${{needs.test-setup.outputs.export == 1}} | |
| steps: | |
| - uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.11' | |
| - 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,torch,transformers,torchvision,onnxruntime] | |
| - name: "🔬 Running export tests" | |
| run: make test TARGETS=export | |