Peter Hawkins 3d4cb9f955 Enable MKL-DNN contraction kernels. ...

Add a new build option --enable_mkl_dnn that enables MKLDNN contraction kernels in XLA. This leads to significant performance improvements for XLA's dot operator. Enable MKL-DNN by default.

Update XLA version to include MKL-DNN build fix.

Also add a new --enable_march_native build option that turns on -march=native. This is unlikely to have a significant performance impact since XLA JIT-compiles most of its code. Leaving this off by default because it also generates code unlikely to run across a wide selection of architectures and so is unsuitable for building pip wheels.

2018-12-14 12:21:25 -05:00

..

jaxlib

Added license header

2018-12-07 12:06:18 -05:00

BUILD

[JAX] Rewrite OSS build script.

2018-11-27 16:51:17 -08:00

build_jaxlib_wheels.sh

improve jaxlib wheel building script

2018-12-09 08:59:08 -08:00

build_wheel_docker_entrypoint.sh

update wheel builder docker entrypoint script

2018-12-07 08:14:24 -08:00

build.py

Enable MKL-DNN contraction kernels.

2018-12-14 12:21:25 -05:00

Dockerfile

update wheel building dockerfile

2018-12-07 08:15:13 -08:00

install_xla_in_source_tree.sh

Split out jax and jaxlib packages (#11)

2018-12-06 21:35:03 -05:00

setup.py

bump jaxlib version for pypi reasons

2018-12-13 18:37:31 -08:00