kornia/.github/workflows/pr_test_cpu.yml at master · YanivHollander/kornia · 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
name: Run tests on CPU

on:
  push:
    branches: [test-me-*]
  pull_request:
    branches: [master]
    types: [opened, reopened, synchronize, ready_for_review, labeled]
  workflow_dispatch:

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

jobs:
  tests-cpu:
    strategy:
      fail-fast: false  # TODO: enable it when CI be stable
      matrix:
        os: ['Ubuntu-latest', 'Windows-latest', 'MacOS-latest']
    uses: ./.github/workflows/tests.yml
    with:
      os: ${{ matrix.os }}
      python-version: '["3.7", "3.10"]'
      pytorch-version: '["1.9.1", "1.13.1"]'
      pytorch-dtype: 'float32'

  tests-cpu-old-torch-new-py:
    strategy:
      fail-fast: false
      matrix:
        os: ['Ubuntu-latest', 'Windows-latest', 'MacOS-latest']

    uses: ./.github/workflows/tests.yml
    with:
      os: ${{ matrix.os }}
      python-version: '["3.9"]'
      pytorch-dtype: 'float32'

  tests-cpu-fp64:
    strategy:
      fail-fast: true
      matrix:
        os: ['Ubuntu-latest']
    uses: ./.github/workflows/tests.yml
    with:
      os: ${{ matrix.os }}
      python-version: '["3.7", "3.10"]'
      pytorch-version: '["1.9.1", "1.13.1"]'
      pytorch-dtype: 'float64'

  tests-nightly:
    if:  contains(github.event.pull_request.labels.*.name, 'nightly')
    name: ${{ matrix.os }} - python-${{ matrix.python-version }}, torch-nightly, ${{ matrix.pytorch-dtype }}

    strategy:
      fail-fast: false
      matrix:
        os: ['Ubuntu-latest', 'Windows-latest'] #, 'MacOS-latest'] add it when https://github.com/pytorch/pytorch/pull/89262 be merged
        pytorch-dtype: ['float32', 'float64']
    uses: ./.github/workflows/tests.yml
    with:
      os: ${{ matrix.os }}
      python-version: '["3.10"]'
      pytorch-version: '["nightly"]'
      pytorch-dtype: ${{ matrix.pytorch-dtype }}
      pytest-extra: '-k dynamo'
  # tests-cpu-half:
  #   strategy:
  #     fail-fast: false # TODO: enable it when CI be stable
  #   uses: ./.github/workflows/tests.yml
  #   with:
  #     os: 'Ubuntu-latest'
  #     python-version: '["3.10"]'
  #     pytorch-version: '["1.13.1"]'
  #     pytorch-dtype: 'float16'
  #     continue-on-error: true