Better handling of torch<2 #79
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: Test with pytest | |
| on: | |
| push: | |
| branches: [ "master" ] | |
| pull_request: | |
| branches: [ "master" ] | |
| permissions: | |
| contents: read | |
| jobs: | |
| test_py38: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Set up python 3.8 | |
| uses: actions/setup-python@v3 | |
| with: | |
| python-version: "3.8" | |
| - uses: actions/checkout@v3 | |
| - name: Install and run tests | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install pytest "jax[cpu]" flax torch tensorflow einops mlx dask tinygrad scipy | |
| pip install . | |
| EINX_FILTER_TRACEBACK=0 pytest | |
| EINX_FILTER_TRACEBACK=0 pytest test/test_invalid_backend.py --noconftest | |
| test_py310: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Set up python 3.10 | |
| uses: actions/setup-python@v3 | |
| with: | |
| python-version: "3.10" | |
| - uses: actions/checkout@v3 | |
| - name: Install and run tests | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install pytest "jax[cpu]" flax dm-haiku torch tensorflow einops equinox mlx dask tinygrad scipy | |
| pip install --upgrade keras | |
| pip install . | |
| EINX_FILTER_TRACEBACK=0 pytest | |
| EINX_FILTER_TRACEBACK=0 pytest test/test_invalid_backend.py --noconftest | |
| test_py310_torchversions: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Set up python 3.10 | |
| uses: actions/setup-python@v3 | |
| with: | |
| python-version: "3.10" | |
| - uses: actions/checkout@v3 | |
| - name: Install and run tests | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install pytest "numpy<2" | |
| pip install . | |
| pip install --upgrade "torch==2.2.0" | |
| EINX_FILTER_TRACEBACK=0 pytest | |
| EINX_FILTER_TRACEBACK=0 pytest test/test_invalid_backend.py --noconftest | |
| pip install --upgrade "torch==2.1.0" | |
| EINX_FILTER_TRACEBACK=0 pytest | |
| EINX_FILTER_TRACEBACK=0 pytest test/test_invalid_backend.py --noconftest | |
| pip install --upgrade "torch==2.0.0" | |
| EINX_FILTER_TRACEBACK=0 pytest | |
| EINX_FILTER_TRACEBACK=0 pytest test/test_invalid_backend.py --noconftest | |
| pip install --upgrade "torch<2" | |
| EINX_FILTER_TRACEBACK=0 pytest test/test_invalid_torch_version.py --noconftest |