auto-round/.github/workflows/compatibility-test.yml at 3cfd83f0acd53bf9206e37b6dce91163feee6a6b · intel/auto-round · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
name: Compatibility Test on ARM64

on:
  pull_request:
    branches: [main]
    types: [opened, reopened, ready_for_review, synchronize]
    paths:
      - "auto_round/**"
      - "auto_round_extension/**"
      - "setup.py"
      - "setup.cfg"
      - "requirements.txt"
      - "requirements-cpu.txt"
      - ".github/workflows/compatibility-test.yml"
      - "!**/*.md"
  workflow_dispatch:

concurrency:
  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
  cancel-in-progress: true

jobs:
  compatibility-test-arm:
    permissions:
      contents: read
      pull-requests: read
    name: ${{ matrix.os-name }} on ARM64
    runs-on: ${{ matrix.os }}
    timeout-minutes: 60
    strategy:
      fail-fast: false
      matrix:
        include:
          - os: macos-latest
            os-name: macOS
            shell: bash
          - os: ubuntu-24.04-arm
            os-name: Linux
            shell: bash

    defaults:
      run:
        shell: ${{ matrix.shell }}

    steps:
      - name: Checkout code
        uses: actions/checkout@v6

      - name: Set up Python
        uses: actions/setup-python@v6
        with:
          python-version: "3.12"

      - name: Verify ARM architecture
        run: |
          python -c "import platform; print(f'Architecture: {platform.machine()}')"

      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip uv setuptools
          uv pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu
          uv pip install datasets --upgrade
          uv build
          uv pip install dist/*.tar.gz && uv pip uninstall auto-round
          uv pip install dist/*.whl
          uv pip list
        env:
          PYTHONUNBUFFERED: "1"
          UV_NO_PROGRESS: "1"
          UV_SYSTEM_PYTHON: "1"
          TQDM_MININTERVAL: "60"

      - name: Run compatibility test
        run: |
          set -e
          python -c "import auto_round; print(auto_round.__version__)"
          echo "============================================================================"
          auto-round --model_name Qwen/Qwen3-0.6B --bits 4 --iters 0 --nsamples 8
          echo "============================================================================"
          auto-round --model_name Qwen/Qwen3-0.6B --bits 4 --iters 1 --nsamples 8
        env:
          TQDM_MININTERVAL: "60"
          PYTHONUNBUFFERED: "1"
          HF_HUB_DISABLE_PROGRESS_BARS: "1"