Peter Hawkins
1abf7cb2dd
Remove -Wno-c++98-c++11-compat directive from jaxlib BUILD file. ( #1544 )
...
We require C++14 now, so the directive is moot.
2019-10-21 11:41:28 -04:00
Skye Wanderman-Milne
796d369efa
Remove licenses() rule comment in BUILD files.
...
Internal tooling doesn't like it.
2019-09-26 14:54:07 -07:00
Peter Hawkins
2725c7e648
Update XLA.
2019-08-09 15:38:20 -04:00
Peter Hawkins
dd10bdba8d
Remove newline from build file.
2019-08-08 16:33:50 -04:00
Peter Hawkins
233598a753
Add newline to build file.
2019-08-08 16:33:04 -04:00
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
5ac356d680
Add support for batched triangular solve and LU decomposition on GPU using cuBlas.
2019-08-08 13:34:53 -04:00
Peter Hawkins
72047c6eca
Update XLA.
2019-08-07 12:55:09 -04:00
Peter Hawkins
6bc476261b
More build formatting fixes.
2019-08-02 13:32:14 -04:00
Peter Hawkins
e0b31ac310
Build formatting fixes.
2019-08-02 13:29:52 -04:00
Peter Hawkins
ed3e2308c1
Add support for linear algebra ops on GPU using Cusolver:
...
* LU decomposition
* Symmetric (Hermitian) eigendecomposition
* Singular value decomposition.
Make LU decomposition tests less sensitive to the exact decomposition; check that we have a decomposition, not precisely the same one scipy returns.
2019-08-02 11:16:15 -04:00
Peter Hawkins
38bffe9a8b
Add a pytreedef.flatten_up_to() method that flattens a PyTree only up to the structure of a PyTreeDef.
...
Make the C++ version of tree_multimap accept tree suffixes of the primary tree. Document and test this behavior.
Remove unnecessary locking in custom node registry; we hold the GIL already so there's no point to the additional locking.
2019-08-01 12:17:00 -04:00
Peter Hawkins
3c3f01e6d3
Address review comments.
2019-07-30 10:15:37 -04:00
Peter Hawkins
510a9167c5
Add C++ implementation of pytree logic.
...
Move jaxlib version test into jax/lib/__init__.py. Make jax/lib mirror the structure of jaxlib; e.g., xla_client is now available as jax.lib.xla_client.
2019-07-29 15:06:05 -04:00
Peter Hawkins
5ceac99d0c
Add newline to the end of jaxlib/BUILD.
2019-04-01 08:23:02 -07:00
Peter Hawkins
5eff830f0e
Move jaxlib version.py into jaxlib, and install it in build/jaxlib as build action.
...
Update jaxlib version check to look in jaxlib.version.
2019-04-01 08:21:22 -07:00
Peter Hawkins
20935448b9
Whitespace fix to jaxlib/BUILD.
2018-12-19 14:51:46 -05:00
Peter Hawkins
3c388b98f1
Add support for calling LAPACK primitives from SciPy from JAX linalg.
2018-12-17 16:30:27 -05:00
