Bump jax from 0.4.14 to 0.4.21 #28
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Tests | |
| on: | |
| pull_request: | |
| push: | |
| branches: | |
| - main | |
| jobs: | |
| test: | |
| name: Run tests | |
| runs-on: ${{ matrix.os }} | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| os: [ubuntu-20.04] | |
| python-version: ["3.7", "3.10"] | |
| backend: [numpy, jax] | |
| env: | |
| PYOM2_DIR: /home/runner/pyom2 | |
| PETSC_VERSION: 3.15 | |
| PETSC_DIR: /home/runner/petsc | |
| PETSC_ARCH: arch-linux-c-opt | |
| OMPI_MCA_rmaps_base_oversubscribe: "1" | |
| OMPI_MCA_mpi_yield_when_idle: "1" | |
| VEROS_REQUIRE_CYTHON_EXT: "1" | |
| steps: | |
| - uses: actions/checkout@v2 | |
| # make sure tags are fetched so we can get a version | |
| - run: git fetch --prune --unshallow --tags | |
| - name: Set up Python ${{ matrix.python-version }} on ${{ matrix.os }} | |
| uses: actions/setup-python@v2 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Restore pip cache | |
| uses: actions/cache@v2 | |
| id: pip-cache | |
| with: | |
| path: ~/.cache/pip | |
| key: ${{ matrix.os }}-pip-${{ matrix.python-version }}-${{ hashFiles('**/requirements*.txt') }} | |
| restore-keys: | | |
| ${{ matrix.os }}-pip- | |
| - name: Restore PyOM2 build cache | |
| uses: actions/cache@v2 | |
| id: pyom2-cache | |
| with: | |
| path: ${{ env.PYOM2_DIR }}/py_src/*.so | |
| key: ${{ matrix.os }}-${{ matrix.python-version }}-${{ hashFiles('vendor/pyom2/patches/*.patch') }}-${{ hashFiles('requirements.txt') }} | |
| - name: Restore PETSc build cache | |
| uses: actions/cache@v2 | |
| id: petsc-cache | |
| with: | |
| path: ${{ env.PETSC_DIR }} | |
| key: ${{ matrix.os }}-${{ env.PETSC_VERSION }} | |
| - name: Setup Python environment | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install wheel | |
| pip install cython | |
| - name: Install system requirements | |
| run: | | |
| sudo apt-get update | |
| sudo apt-get install libopenmpi-dev | |
| - name: Build PETSc | |
| if: steps.petsc-cache.outputs.cache-hit != 'true' | |
| run: | | |
| git clone -b v$PETSC_VERSION --depth 1 https://gitlab.com/petsc/petsc.git $PETSC_DIR | |
| pushd $PETSC_DIR | |
| ./configure --with-debugging=0 -with-shared-libraries --with-precision=double | |
| make all | |
| popd | |
| - name: Install Veros | |
| run: | | |
| pip install mpi4py | |
| if [ ${{ matrix.backend }} == "jax" ] | |
| then | |
| pip install mpi4jax | |
| pip install -e .[test,jax] | |
| else | |
| pip install -e .[test] | |
| fi | |
| pip install petsc4py==$PETSC_VERSION --no-deps | |
| # Build PyOM2 after Veros to make sure we have compatible versions of NumPy / f2py | |
| - name: Build PyOM2 | |
| if: steps.pyom2-cache.outputs.cache-hit != 'true' | |
| run: | | |
| mkdir -p $PYOM2_DIR | |
| cp -r vendor/pyom2/* $PYOM2_DIR | |
| pushd $PYOM2_DIR | |
| mv pyOM2_site_specific site_specific.mk_ | |
| tar xzf pyOM2.1.0.tar.gz | |
| git init | |
| for patchfile in ./patches/*.patch; do | |
| git apply --whitespace=fix $patchfile | |
| done | |
| make -C py_src -j 4 > /dev/null | |
| popd | |
| - name: Export paths | |
| run: | | |
| echo "PYOM2_LIB=$(readlink -f $PYOM2_DIR/py_src/pyOM_code.*.so)" >> $GITHUB_ENV | |
| echo "PYOM2_LIB_MPI=$(readlink -f $PYOM2_DIR/py_src/pyOM_code_MPI.*.so)" >> $GITHUB_ENV | |
| - name: Run tests | |
| run: | | |
| pytest . -v --cov --pyom2-lib $PYOM2_LIB --backend ${{ matrix.backend }} | |
| - name: Upload coverage | |
| uses: codecov/codecov-action@v1 |