pr_test_cpu.yml

name: Tests on CPU (PR)

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

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

jobs:
  tests-cpu:
    strategy:
      fail-fast: true
      matrix:
        os: ['Ubuntu-latest', 'Windows-latest', 'MacOS-latest']
        pytorch-dtype: ['float32', 'float64']
        exclude:
          - os: 'Windows-latest'
            pytorch-dtype: 'float64'
          - os: 'MacOS-latest'
            pytorch-dtype: 'float64'

    uses: kornia/workflows/.github/workflows/tests.yml@v1.8.1
    with:
      os: ${{ matrix.os }}
      python-version: '["3.8", "3.11"]'
      pytorch-version: '["1.9.1", "2.2.2"]'
      pytorch-dtype: ${{ matrix.pytorch-dtype }}

  coverage:
    uses: kornia/workflows/.github/workflows/coverage.yml@v1.8.1

  typing:
    uses: kornia/workflows/.github/workflows/mypy.yml@v1.8.1

  tutorials:
    uses: kornia/workflows/.github/workflows/tutorials.yml@v1.8.1

  docs:
    uses: kornia/workflows/.github/workflows/docs.yml@v1.8.1

  collector:
    needs: [coverage, tests-cpu, tutorials, typing, docs]
    if: always()
    runs-on: ubuntu-latest
    steps:
    - name: check for failures
      if: contains(needs.*.result, 'failure') || contains(needs.*.result, 'cancelled')
      run: echo job failed && exit 1

  tests-nightly:
    if:  contains(github.event.pull_request.labels.*.name, 'nightly')
    name: ${{ matrix.os }} - 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: kornia/workflows/.github/workflows/tests.yml@v1.8.1
    with:
      os: ${{ matrix.os }}
      pytorch-version: '["nightly"]'
      pytorch-dtype: ${{ matrix.pytorch-dtype }}