167 Commits

Author SHA1 Message Date
Peter Hawkins
fef315b6e6 Add ability to pass extra bazel options to build script.
Remove cublas/cusolver dependencies from Jaxlib python code.
2019-08-08 16:14:45 -04:00
Peter Hawkins
40c517e9ef Add --spawn_strategy from TensorFlow configuration. 2019-08-01 21:43:03 -04:00
Peter Hawkins
fafe69bd8f Relax Bazel maximum version check.
Remove debug print from lax_numpy_test.py.
2019-05-29 19:38:55 -04:00
Peter Hawkins
8cbc475a8b Add "bazel shutdown" to the build script.
Fixes #739.
2019-05-20 20:53:44 -04:00
Peter Hawkins
5599f473f0 Revert to requiring Bazel 0.24 for JAX builds.
It turns out TF does not fully support 0.25 yet.

Add a check for a maximum bazel version to the build script.

Fixes #702.
2019-05-12 09:25:10 -04:00
Peter Hawkins
7f9e1809bb Fix build with Bazel 0.25.
Update minimum Bazel version to 0.25.0.
2019-05-10 11:53:15 -04:00
Peter Hawkins
f60d927df8 Update XLA version to include CUDA version check fix and newer XLA.
Make specification of CUDA environment variables optional.

Fixes #627.
Fixes #276, although the fix requires a new Jaxlib release.
2019-04-25 16:19:53 -07:00
Peter Hawkins
e8638f73cc Update XLA.
Switches XLA Python bindings to use pybind11.
Update XRT support to point to newer XRT client.
Update minimum bazel version to 0.24.0.

Fix missing backend argument to XLA Compile() calls.
2019-03-31 10:56:47 -07:00
Peter Hawkins
f8eda9bb64 Set default Apple platform to fix TensorFlow build breakage when built for Jaxlib. 2019-03-26 22:26:12 -04:00
Matthew Johnson
ab0c0585da update build file to work with cuda nccl 2019-03-08 08:57:55 -08:00
Peter Hawkins
17fbdcaa84 Update default bazel release in build script to 0.22.0.
Update imported rules to a version that supports newer Bazel versions.
2019-03-02 15:10:42 -05:00
Peter Hawkins
5031016465 Disable a number of TF features in build.py that we don't use and don't need to build. 2019-01-07 10:59:08 -05:00
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
Peter Hawkins
cad36945a2 Update build.py to write its .bazelrc in the root directory of the repository, next to the workspace.
Fixes build breakage caused by workspace move.
2018-12-12 15:55:55 -05:00
Peter Hawkins
5510ceaf55 Set --distinct_host_configuration=false in the bazel options.
This makes initial builds cheaper (since we don't need to build some files in separate host and target configurations) but may make switching between build configurations more expensive (since we can share less work). The build script should optimize for the former.
2018-12-10 09:18:49 -05:00
Matthew Johnson
d4f32dff7b update build script for split packages 2018-12-07 07:34:37 -08:00
Matthew Johnson
bbc92ce6eb
Split out jax and jaxlib packages (#11)
factor out 'jaxlib' as separate package
2018-12-06 21:35:03 -05:00