build.yml

name: build

on:
  pull_request:
  push:
    branches:
      - master
  workflow_dispatch:

concurrency:
  # Skip intermediate builds: always.
  # Cancel intermediate builds: only if it is a pull request build.
  group: ${{ github.workflow }}-${{ github.ref }}
  cancel-in-progress: ${{ startsWith(github.ref, 'refs/pull/') }}

jobs:
  build:
    runs-on: ${{ matrix.os }}
    strategy:
      fail-fast: false
      matrix:
        python-version: ["3.8", "3.9", "3.10"]
        os: [ubuntu-latest, macos-latest, windows-latest]

    steps:
      - uses: actions/checkout@v3

      - uses: actions/setup-python@v4
        with:
          python-version: ${{ matrix.python-version }}

      - name: Install DeepXDE with dev requirements
        run: python -m pip install -e ".[dev]"

      - name: List environment
        run: python -m pip freeze

      - name: Build with TensorFlow v1
        env:
          DDE_BACKEND: tensorflow.compat.v1
        run: |
          python -m pip install tensorflow
          python -c "import deepxde"

      - name: Build with TensorFlow
        env:
          DDE_BACKEND: tensorflow
        run: |
          python -m pip install tensorflow-probability
          python -c "import deepxde"

      - name: Build with PyTorch
        env:
          DDE_BACKEND: pytorch
        run: |
          python -m pip install torch
          python -c "import deepxde"

      - name: Build with Jax
        if: matrix.os != 'windows-latest'  # jaxlib is not available on windows
        env:
          DDE_BACKEND: jax
        run: |
          python -m pip install jax flax optax
          python -c "import deepxde"

      - name: Build with Paddle
        env:
          DDE_BACKEND: paddle
          ubuntu-latest: "https://www.paddlepaddle.org.cn/whl/linux/cpu-mkl/develop.html"
          macos-latest: "https://www.paddlepaddle.org.cn/whl/mac/cpu/develop.html"
          windows-latest: "https://www.paddlepaddle.org.cn/whl/windows/cpu-mkl-avx/develop.html"
        run: |
          python -m pip install paddlepaddle==0.0.0 -f ${{ env.${{ matrix.os }} }}
          python -c "import deepxde"