mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 01:06:04 +00:00
45 lines
1.6 KiB
YAML
45 lines
1.6 KiB
YAML
sudo: false
|
|
notifications:
|
|
email: false
|
|
dist: xenial
|
|
language: python
|
|
python:
|
|
- "3.6"
|
|
env:
|
|
- JAX_ENABLE_X64=0 JAX_NUM_GENERATED_CASES=25
|
|
- JAX_ENABLE_X64=1 JAX_NUM_GENERATED_CASES=25
|
|
matrix:
|
|
include:
|
|
- python: "3.7"
|
|
env: JAX_ENABLE_X64=1 JAX_ONLY_DOCUMENTATION=true
|
|
|
|
before_install:
|
|
- wget https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh;
|
|
- bash miniconda.sh -b -p $HOME/miniconda
|
|
- export PATH="$HOME/miniconda/bin:$PATH"
|
|
- conda config --set always_yes yes --set changeps1 no --set show_channel_urls yes
|
|
- conda config --set channel_priority strict
|
|
- conda config --set add_pip_as_python_dependency yes
|
|
- conda config --remove channels defaults
|
|
- conda config --add channels conda-forge
|
|
- conda update -q conda
|
|
install:
|
|
- conda install --yes python=$TRAVIS_PYTHON_VERSION pip six protobuf>=3.6.0 absl-py opt_einsum numpy scipy pytest-xdist pytest-benchmark fastcache
|
|
# The jaxlib version should match the minimum jaxlib version in
|
|
# jax/lib/__init__.py. This tests JAX PRs against the oldest permitted
|
|
# jaxlib.
|
|
- pip install jaxlib==0.1.37
|
|
- pip install -v .
|
|
# The following are needed to test the Colab notebooks and the documentation building
|
|
- if [[ "$JAX_ONLY_DOCUMENTATION" != "" ]]; then
|
|
conda install --yes -c conda-forge pandoc ipykernel;
|
|
conda install --yes sphinx sphinx_rtd_theme nbsphinx jupyter_client matplotlib;
|
|
pip install sklearn;
|
|
fi
|
|
script:
|
|
- if [[ "$JAX_ONLY_DOCUMENTATION" == "" ]]; then
|
|
pytest -n 1 tests examples -W ignore ;
|
|
else
|
|
sphinx-build -b html -D nbsphinx_execute=always docs docs/build/html;
|
|
fi
|