mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-23 20:56:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/.travis.yml

55 lines
2.0 KiB
YAML
Raw Normal View History

add initial travis ci file
2018-12-13 13:39:34 -08:00
sudo: false
notifications:
email: false
use 'dist: xenial' in travis.yml to get updated toolchain
2018-12-13 13:53:41 -08:00
dist: xenial
add initial travis ci file
2018-12-13 13:39:34 -08:00
language: python
python:
- "2.7"
- "3.6"
env:
reduce travis sampled test cases from 100 to 25 This change is just to speed up travis, which has ballooned from ~10 minutes to ~40 minutes.
2019-07-02 12:38:16 -07:00
- JAX_ENABLE_X64=0 JAX_NUM_GENERATED_CASES=25
- JAX_ENABLE_X64=1 JAX_NUM_GENERATED_CASES=25
Add a separate build matrix entry for documentation testing. (#1495) * Add a separate build matrix entry for documentation testing. This way we parallelize the unit tests with the documentation tests.
2019-10-13 09:34:51 +02:00
matrix:
include:
- python: "3.7"
env: JAX_ENABLE_X64=1 JAX_ONLY_DOCUMENTATION=true
add multi_device_test in travis
2019-12-18 11:20:42 -08:00
include:
- python: "3.7"
env: MULTI_DEVICE_TEST=1
Add a separate build matrix entry for documentation testing. (#1495) * Add a separate build matrix entry for documentation testing. This way we parallelize the unit tests with the documentation tests.
2019-10-13 09:34:51 +02:00
add initial travis ci file
2018-12-13 13:39:34 -08:00
before_install:
- if [[ "$TRAVIS_PYTHON_VERSION" == "2.7" ]]; then
fix broken travis path from previous commit
2019-06-07 16:00:11 -07:00
wget https://repo.continuum.io/miniconda/Miniconda2-latest-Linux-x86_64.sh -O miniconda.sh;
add initial travis ci file
2018-12-13 13:39:34 -08:00
else
fix broken travis path from previous commit
2019-06-07 16:00:11 -07:00
wget https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh;
add initial travis ci file
2018-12-13 13:39:34 -08:00
fi
- bash miniconda.sh -b -p $HOME/miniconda
- export PATH="$HOME/miniconda/bin:$PATH"
try following jupyterlab/jupyter-renderers#212 to fix travis
2019-10-15 23:07:31 +00:00
- 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
add initial travis ci file
2018-12-13 13:39:34 -08:00
- conda config --add channels conda-forge
try following jupyterlab/jupyter-renderers#212 to fix travis
2019-10-15 23:07:31 +00:00
- conda update -q conda
add initial travis ci file
2018-12-13 13:39:34 -08:00
install:
Support IntEnum values as arguments to JAX functions. (#1840) * Support IntEnum values as arguments to JAX functions. When abstractifying a Python value, search the method-resolution order (MRO) of the type rather than only looking at the value's own type. IntEnum instances are subclasses of int, so this allows us to correctly handle them as integers, much as NumPy itself does.
2019-12-11 12:27:11 -05:00
- conda install --yes python=$TRAVIS_PYTHON_VERSION pip six protobuf>=3.6.0 absl-py opt_einsum numpy scipy pytest-xdist pytest-benchmark fastcache
Pin the minimum jaxlib version in travis.yml. (#1767)
2019-11-26 10:24:06 -05:00
# The jaxlib version should match the minimum jaxlib version in
# jax/lib/__init__.py. This tests JAX PRs against the oldest permitted
# jaxlib.
Make pmap properly replicate closed-over constants. (#1847) With this change, a value `x` can be replicated `nrep` times as follows: ```python pmap(lambda _: x)(np.arange(nrep)) ``` This will broadcast `x` into a ShardedDeviceArray suitable for passing into another pmap with the same input shape. If `x` will be passed into a pmap with `devices` or a nested pmap, the replication pmap(s) should follow that structure. For example: ```python x = pmap(pmap(lambda _: x))(np.ones(2, 4)) pmap(pmap(lambda i: i**2 + x))(np.ones(2, 4)) ```
2019-12-17 16:22:55 -08:00
- pip install jaxlib==0.1.37
add initial travis ci file
2018-12-13 13:39:34 -08:00
- pip install -v .
Switched testing to using sphinx (part of the documentation generation)
2019-10-02 21:44:14 +02:00
# The following are needed to test the Colab notebooks and the documentation building
Add a separate build matrix entry for documentation testing. (#1495) * Add a separate build matrix entry for documentation testing. This way we parallelize the unit tests with the documentation tests.
2019-10-13 09:34:51 +02:00
- if [[ "$JAX_ONLY_DOCUMENTATION" != "" ]]; then
conda install --yes -c conda-forge pandoc ipykernel;
conda install --yes sphinx sphinx_rtd_theme nbsphinx jupyter_client matplotlib;
Add sklearn to Travis, for documentation building. (#1547) * Add sklearn to Travis, for documentation building. * Add score_matching to auto-built notebooks
2019-10-21 23:24:16 +02:00
pip install sklearn;
Add a separate build matrix entry for documentation testing. (#1495) * Add a separate build matrix entry for documentation testing. This way we parallelize the unit tests with the documentation tests.
2019-10-13 09:34:51 +02:00
fi
add initial travis ci file
2018-12-13 13:39:34 -08:00
script:
make op-by-op computation follow arg placement
2019-12-17 17:49:06 -08:00
- if [[ "$MULTI_DEVICE_TEST" != "" ]]; then
XLA_FLAGS=--xla_force_host_platform_device_count=2 python tests/api_test.py MultiDeviceTest;
elif [[ "$JAX_ONLY_DOCUMENTATION" == "" ]]; then
Add a separate build matrix entry for documentation testing. (#1495) * Add a separate build matrix entry for documentation testing. This way we parallelize the unit tests with the documentation tests.
2019-10-13 09:34:51 +02:00
pytest -n 1 tests examples -W ignore ;
else
sphinx-build -b html -D nbsphinx_execute=always docs docs/build/html;
Added support for testing Colab notebooks. Testing is done by running "jupyter nbconvert --to notebook" and then parsing the resulting notebook to look for errors. One can declare expected errors, and the test will fail if those are missing. In the process of doig this, found and fixed a bug in the autodiff_cookbook notebook.
2019-10-02 17:51:00 +02:00
fi
Reference in New Issue Copy Permalink