7 Commits

Author SHA1 Message Date
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