Skip to content

Refactor linear dispatch to use new torch kernels #108

Refactor linear dispatch to use new torch kernels

Refactor linear dispatch to use new torch kernels #108

Workflow file for this run

name: Python examples (CPU)
on:
push:
branches:
- main
paths:
- "optimum/quanto/**"
- "examples/**"
- "pyproject.toml"
pull_request:
types: [assigned, opened, synchronize, reopened]
paths:
- "optimum/quanto/**"
- "examples/**"
- "pyproject.toml"
jobs:
conventional-commits:
uses: ./.github/workflows/conventional-commits.yml
python-quality:
uses: ./.github/workflows/python-quality.yml
run-examples:
needs: [conventional-commits, python-quality]
runs-on: ubuntu-latest
strategy:
fail-fast: false
matrix:
python-version: ["3.11"]
steps:
- uses: actions/checkout@v2
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v2
with:
python-version: ${{ matrix.python-version }}
# Install dependencies
- name: Configure and install dependencies
run: |
pip install --upgrade pip
pip install .[examples]
# Run examples
- name: Run MNIST examples
run: |
for w in int4 int8 float8; do \
for a in none int8 float8; do \
python examples/vision/image-classification/mnist/quantize_mnist_model.py \
--weights $w --activations $a; \
done; \
done
- name: Run OWL detection examples
run: |
for w in int4 int8 float8; do \
python examples/vision/object-detection/quantize_owl_model.py \
--image http://images.cocodataset.org/val2017/000000039769.jpg \
--texts "a photo of a cat" "a remote" \
--weights $w; \
done
- name: Run text-classification examples
run: |
for w in int4 int8; do \
for a in none int8; do \
python examples/nlp/text-classification/sst2/quantize_sst2_model.py \
--weights $w --activations $a; \
done; \
done