From cb66d7a3cbca9db9299320b826bf669d8b13bdd7 Mon Sep 17 00:00:00 2001
From: David Corvoysier <david@huggingface.co>
Date: Tue, 2 Apr 2024 10:08:05 +0200
Subject: [PATCH] ci: add examples workflow

---
 .github/workflows/python-examples.yml | 58 +++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 .github/workflows/python-examples.yml

diff --git a/.github/workflows/python-examples.yml b/.github/workflows/python-examples.yml
new file mode 100644
index 00000000..5ef3e888
--- /dev/null
+++ b/.github/workflows/python-examples.yml
@@ -0,0 +1,58 @@
+name: Python examples (CPU)
+
+on:
+  push:
+    branches:
+      - main
+    paths:
+      - "quanto/**"
+      - "examples/**"
+      - "pyproject.toml"
+      - "setup.py"
+  pull_request:
+    types: [assigned, opened, synchronize, reopened]
+    paths:
+      - "quanto/**"
+      - "examples/**"
+      - "pyproject.toml"
+      - "setup.py"
+
+jobs:
+  build-ubuntu:
+    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 .[dev] torchvision transformers
+
+      # 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 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
+