heat/.github/workflows/push_main.yml at 0e8f20c039009909991efa22d12403a2a49319b3 · brownbaerchen/heat · 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
118
119
name: On Push Main
on:
  push:
    branches:
      - main
      - stable
    paths:
      - 'heat/**'
      - 'tests/**'
      - 'pyproject.toml'
jobs:
  codebase-test-and-bench-main:
    name: Codebase Tests and Benchmarks
    if: ${{ !github.event.repository.fork }}
    runs-on: ubuntu-latest
    steps:
      - name: Harden the runner (Audit all outbound calls)
        uses: step-security/harden-runner@9af89fc71515a100421586dfdb3dc9c984fbf411 # v2.19.4
        with:
          egress-policy: audit
      - name: Checkout code
        uses: actions/checkout@df4cb1c069e1874edd31b4311f1884172cec0e10 # v6.0.3
        with:
          ref: ${{ github.sha }}
      - uses: ./.github/actions/trigger_codebase_tests
        with:
          branch: "main"
          sha: ${{ github.sha}}
          trigger_token: ${{ secrets.CX_TRIGGER_TOKEN }}
      - uses: ./.github/actions/trigger_codebase_benchmarks
        with:
          branch: "main"
          sha: ${{ github.sha}}
          author: 'heat_team'
          trigger_token: ${{ secrets.CX_TRIGGER_TOKEN }}
  gh-matrix-tests:
    runs-on: ubuntu-latest
    timeout-minutes: 120
    strategy:
      fail-fast: false
      matrix:
        py-version:
          - '3.11'
          - '3.12'
          - '3.13'
          - '3.14'
        mpi: [ 'mpich' ]
        pytorch-version:
          - 'torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1'
          - 'torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1'
          - 'torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1'
          - 'torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0'
          - 'torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1'
          - 'torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0'
          - 'torch==2.9.1 torchvision==0.24.1 torchaudio==2.9.1'
          - 'torch==2.10.0 torchvision==0.25.0 torchaudio==2.10.0'
          - 'torch==2.11.0 torchvision==0.26.0 torchaudio==2.11.0'
        exclude:
          - py-version: '3.14'
            pytorch-version: 'torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1'
          - py-version: '3.14'
            pytorch-version: 'torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1'
          - py-version: '3.14'
            pytorch-version: 'torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1'
          - py-version: '3.14'
            pytorch-version: 'torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0'
          - py-version: '3.14'
            pytorch-version: 'torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1'
          - py-version: '3.14'
            pytorch-version: 'torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0'
          - py-version: '3.13'
            pytorch-version: 'torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1'
          - py-version: '3.13'
            pytorch-version: 'torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1'
          - py-version: '3.13'
            pytorch-version: 'torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1'

    name: Python ${{ matrix.py-version }} with ${{ matrix.pytorch-version }}
    steps:
      - name: Harden Runner
        uses: step-security/harden-runner@9af89fc71515a100421586dfdb3dc9c984fbf411 # v2.19.4
        with:
          egress-policy: audit
      - name: Checkout code
        uses: actions/checkout@df4cb1c069e1874edd31b4311f1884172cec0e10 # v6.0.3
        with:
          ref: ${{ github.sha }}
      - uses: ./.github/actions/gh_tests
        with:
          mode: "full"
          py-version: ${{ matrix.py-version }}
          mpi: ${{ matrix.mpi }}
          pytorch-version: ${{ matrix.pytorch-version }}
  check_reuse_compliance:
    runs-on: ubuntu-latest
    timeout-minutes: 120
    name: Check REUSE compliance
    steps:
      - name: Harden Runner
        uses: step-security/harden-runner@a5ad31d6a139d249332a2605b85202e8c0b78450 # v2.19.1
        with:
          egress-policy: audit

      - name: Checkout
        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2

      - name: Setup Python
        uses: actions/setup-python@a309ff8b426b58ec0e2a45f0f869d46889d02405 # v6.2.0
        with:
          python-version: '3.14'
          architecture: x64

      - name: Install REUSE
        run: |
          pip install reuse

      - name: Run linter
        run: |
          reuse lint