Skip to content
ci

feat: QAT (quantization-aware training) variant #5

Sign in to view logs

feat: QAT (quantization-aware training) variant

feat: QAT (quantization-aware training) variant #5

Workflow file for this run

.github/workflows/ci.yml at b3fcb99
name: ci
on:
push:
branches: [main]
pull_request:
branches: [main]
permissions:
contents: read
jobs:
lint:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.11"
cache: pip
- name: Install lint tools
run: |
python -m pip install --upgrade pip
python -m pip install ruff==0.4.10 black==24.4.2
- name: Ruff
run: ruff check src tests
- name: Black
run: black --check src tests
typecheck:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.11"
cache: pip
- name: Install
run: |
python -m pip install --upgrade pip
pip install --index-url https://download.pytorch.org/whl/cpu torch==2.2.2 torchvision==0.17.2
pip install -e ".[dev]"
- name: Mypy
run: mypy src/quant_explorer
test:
runs-on: ubuntu-latest
strategy:
fail-fast: false
matrix:
python-version: ["3.11", "3.12"]
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
cache: pip
- name: Install
run: |
python -m pip install --upgrade pip
pip install --index-url https://download.pytorch.org/whl/cpu torch==2.2.2 torchvision==0.17.2
pip install -e ".[dev]"
- name: Unit tests + coverage gate (>= 56%)
run: pytest tests/unit -v --cov=src/quant_explorer --cov-report=term
integration-tiny:
runs-on: ubuntu-latest
needs: [lint, test]
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.11"
cache: pip
- name: Install
run: |
python -m pip install --upgrade pip
pip install --index-url https://download.pytorch.org/whl/cpu torch==2.2.2 torchvision==0.17.2
pip install -e ".[dev]"
- name: Cache CIFAR-10 dataset
uses: actions/cache@v4
with:
path: data
key: cifar10-${{ hashFiles('src/quant_explorer/data.py') }}
restore-keys: cifar10-
- name: Tiny pipeline
env:
RUN_INTEGRATION: "1"
run: pytest tests/integration -v
bench-smoke:
runs-on: ubuntu-latest
needs: [lint, test]
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.11"
cache: pip
- name: Install
run: |
python -m pip install --upgrade pip
pip install --index-url https://download.pytorch.org/whl/cpu torch==2.2.2 torchvision==0.17.2
pip install -e ".[dev]"
- name: Bench fp32_baseline (loads committed weights)
run: |
quant-explorer bench --config fp32_baseline --warmup 2 --iters 20
- name: Validate result JSON shape
run: |
python - <<'PY'
import json, pathlib
p = pathlib.Path("artifacts/results/fp32_baseline.json")
assert p.exists(), p
data = json.loads(p.read_text())
assert "size" in data
assert "latency" in data and len(data["latency"]) > 0
assert "memory" in data
for lat in data["latency"]:
assert {"batch_size", "p50_ms", "p95_ms", "p99_ms"}.issubset(lat)
print("ok")
PY
qat-smoke:
runs-on: ubuntu-latest
needs: [lint, test]
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.11"
cache: pip
- name: Install
run: |
python -m pip install --upgrade pip
pip install --index-url https://download.pytorch.org/whl/cpu torch==2.2.2 torchvision==0.17.2
pip install -e ".[dev]"
- name: Cache CIFAR-10 dataset
uses: actions/cache@v4
with:
path: data
key: cifar10-${{ hashFiles('src/quant_explorer/data.py') }}
restore-keys: cifar10-
- name: QAT fine-tune (tiny — 256 train images, 1 epoch)
run: |
quant-explorer qat-finetune --epochs 1 --train-subset 256 --batch-size 64
- name: Bench QAT graph
run: quant-explorer bench --config qat_int8 --warmup 2 --iters 20
- name: Validate QAT result JSON
run: |
python - <<'PY'
import json, pathlib
p = pathlib.Path("artifacts/results/qat_int8.json")
assert p.exists(), p
data = json.loads(p.read_text())
assert data["size"]["kb"] > 0
assert all(lat["p50_ms"] > 0 for lat in data["latency"])
print("ok")
PY
multi-bench-regress:
runs-on: ubuntu-latest
needs: [lint, test]
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.11"
cache: pip
- name: Install
run: |
python -m pip install --upgrade pip
pip install --index-url https://download.pytorch.org/whl/cpu torch==2.2.2 torchvision==0.17.2
pip install -e ".[dev]"
- name: Run multi-model bench (random init, no accuracy)
run: quant-explorer multi-bench --warmup 1 --iters 5
- name: Bench-regress structural gate
run: python scripts/bench_regress_check.py
build:
runs-on: ubuntu-latest
needs: [test]
steps:
- uses: actions/checkout@v4
- name: Build image
run: docker build -t quant-explorer:ci .
- name: Smoke run inside container
run: |
# ENTRYPOINT is `quant-explorer`; --help is the safe smoke arg.
docker run --rm quant-explorer:ci --help | grep -q "Quant-explorer CLI"
# Override entrypoint to confirm the package imports cleanly.
docker run --rm --entrypoint python quant-explorer:ci -c "from quant_explorer.cli import main; print('ok')"