ao/.github/workflows/regression_test.yml at b764ffbc15992a0a9b64201780e7f56c58812a30 · pytorch/ao · 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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
name: Run Regression Tests

on:
  push:
    branches:
      - main
      - 'gh/**'
  pull_request:
    branches:
      - main
      - 'gh/**'

concurrency:
  group: regression_test-${{ github.workflow }}-${{ github.ref == 'refs/heads/main' && github.run_number || github.ref }}
  cancel-in-progress: true

env:
  HF_TOKEN: ${{ secrets.HF_TOKEN }}

jobs:
  test-nightly:
    strategy:
      fail-fast: false
      matrix:
        include:
          - name: CUDA Nightly
            runs-on: linux.g5.12xlarge.nvidia.gpu
            torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu126'
            gpu-arch-type: "cuda"
            gpu-arch-version: "12.6"
          - name: CPU Nightly
            runs-on: linux.4xlarge
            torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cpu'
            gpu-arch-type: "cpu"
            gpu-arch-version: ""

    permissions:
      id-token: write
      contents: read
    uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
    with:
      timeout: 180
      runner: ${{ matrix.runs-on }}
      gpu-arch-type: ${{ matrix.gpu-arch-type }}
      gpu-arch-version: ${{ matrix.gpu-arch-version }}
      submodules: recursive
      script: |
        conda create -n venv python=3.10 libgcc-ng=11.2.0 libstdcxx-ng=11.2.0 -y
        conda activate venv
        python -m pip install --upgrade pip
        pip install ${{ matrix.torch-spec }}
        pip install -r dev-requirements.txt
        pip install . --no-build-isolation
        export CONDA=$(dirname $(dirname $(which conda)))
        export LD_LIBRARY_PATH=$CONDA/lib/:$LD_LIBRARY_PATH
        pytest test --verbose -s
  test:
    strategy:
      fail-fast: false
      matrix:
        include:
          - name: CUDA 2.8
            runs-on: linux.g5.12xlarge.nvidia.gpu
            torch-spec: 'torch==2.8.0'
            gpu-arch-type: "cuda"
            gpu-arch-version: "12.6"
            dev-requirements-overrides: ""
          - name: CUDA 2.9
            runs-on: linux.g5.12xlarge.nvidia.gpu
            torch-spec: 'torch==2.9.1'
            gpu-arch-type: "cuda"
            gpu-arch-version: "12.6"
            dev-requirements-overrides: ""
          - name: CUDA 2.10
            runs-on: linux.g5.12xlarge.nvidia.gpu
            torch-spec: 'torch==2.10.0'
            gpu-arch-type: "cuda"
            gpu-arch-version: "12.6"
            dev-requirements-overrides: ""

          - name: CPU 2.8
            runs-on: linux.4xlarge
            torch-spec: 'torch==2.8.0 --index-url https://download.pytorch.org/whl/cpu'
            gpu-arch-type: "cpu"
            gpu-arch-version: ""
            dev-requirements-overrides: ""
          - name: CPU 2.9
            runs-on: linux.4xlarge
            torch-spec: 'torch==2.9.1 --index-url https://download.pytorch.org/whl/cpu'
            gpu-arch-type: "cpu"
            gpu-arch-version: ""
            dev-requirements-overrides: ""
          - name: CPU 2.10
            runs-on: linux.4xlarge
            torch-spec: 'torch==2.10.0 --index-url https://download.pytorch.org/whl/cpu'
            gpu-arch-type: "cpu"
            gpu-arch-version: ""
            dev-requirements-overrides: ""

    uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
    with:
      timeout: 180
      runner: ${{ matrix.runs-on }}
      gpu-arch-type: ${{ matrix.gpu-arch-type }}
      gpu-arch-version: ${{ matrix.gpu-arch-version }}
      submodules: recursive
      script: |
        conda create -n venv python=3.10 libgcc-ng=11.2.0 libstdcxx-ng=11.2.0 -y
        conda activate venv
        python -m pip install --upgrade pip
        pip install ${{ matrix.torch-spec }}
        sed -i '${{ matrix.dev-requirements-overrides }}' dev-requirements.txt
        pip install -r dev-requirements.txt
        pip install . --no-build-isolation
        export CONDA=$(dirname $(dirname $(which conda)))
        export LD_LIBRARY_PATH=$CONDA/lib/:$LD_LIBRARY_PATH
        pytest test --verbose -s