heat/.github/workflows/ci_full.yaml at main · helmholtz-analytics/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
name: ci_full

on:
  push:
    branches:
      - main
      - stable
    paths:
      - 'heat/**'
      - 'pyproject.toml'
      - '.github/workflows/ci_full.yaml'

jobs:
  full_matrix_test:
    runs-on: ubuntu-latest
    timeout-minutes: 120
    strategy:
      fail-fast: false
      matrix:
        py-version:
          - '3.10'
          - '3.11'
          - '3.12'
          - '3.13'
          - '3.14'
        mpi: [ 'openmpi' ]
        install-options: [ '.', '.[hdf5,netcdf,pandas,zarr]' ]
        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'
        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'
          - py-version: '3.10'
            install-options: '.[hdf5,netcdf,pandas,zarr]'

    name: Python ${{ matrix.py-version }} with ${{ matrix.pytorch-version }}; options ${{ matrix.install-options }}
    steps:
      - name: Harden Runner
        uses: step-security/harden-runner@a90bcbc6539c36a85cdfeb73f7e2f433735f215b # v2.15.0
        with:
          egress-policy: audit

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

      - name: Setup MPI
        uses: mpi4py/setup-mpi@3969f247e8fceef153418744f9d9ee6fdaeda29f # v1.2.0
        with:
          mpi: ${{ matrix.mpi }}

      - name: Use Python ${{ matrix.py-version }}
        uses: actions/setup-python@83679a892e2d95755f2dac6acb0bfd1e9ac5d548 # v6.1.0
        with:
          python-version: ${{ matrix.py-version }}
          architecture: x64

      - name: Test
        run: |
          pip install pytest
          pip install ${{ matrix.pytorch-version }} ${{ matrix.install-options }} --extra-index-url https://download.pytorch.org/whl/cpu
          mpirun -n 3 pytest heat/
          mpirun -n 4 pytest heat/