17 Commits

Author SHA1 Message Date
Peter Hawkins
1fc9afd03a Add support for running JAX tests under Bazel.
This is an alternative method for running the tests that some users may prefer: pytest is and will remain fully supported.

To use this, one creates a .bazelrc by running the existing `build.py` script, and then one can run the tests by running:
```
bazel test -c opt //tests/...
```

Issue #7323

PiperOrigin-RevId: 458551208
2022-07-01 15:07:22 -07:00
Roy Frostig
28828772b8 remove unuzed bazel build rules, including bazel test definitions 2018-12-16 10:42:50 -08:00
Roy Frostig
a97f820961 update examples/BUILD 2018-12-15 15:54:31 -08:00
Peter Hawkins
3561b432c2 Add Cholesky, QR, and Triangular solve implementations.
* Adds lax.{cholesky,triangular_solve,qr}. Adds a JVP for Cholesky.
* Adds a transpose rule for add_p, needed by the Cholesky JVP.
* Adds np.linalg.{cholesky,qr,dot,matmul,trace}.
* Adds scipy.linalg.{cholesky,qr,solve_triangular,tril,triu}.

Pair programmed with mattjj.
2018-12-13 13:03:08 -05:00
Roy Frostig
cfc37eb0be add a test of the resnet50 example 2018-12-12 16:09:20 -08:00
Roy Frostig
9c69014318 Only build-depend on libjax optionally (via bazel CLI flag) in the generated bazel test targets. 2018-12-12 18:01:44 -05:00
Dougal Maclaurin
c3374a9d5f Added build rule for generated_fun_test (formerly quickish_check) 2018-12-06 17:04:00 -05:00
Dougal Maclaurin
29113dd606 Made tests runnable with bazel 2018-12-06 17:00:47 -05:00
Roy Frostig
f5b051b431 Double gpu test shards
PiperOrigin-RevId: 223647959
2018-12-02 11:50:46 -08:00
Roy Frostig
20878c76f4 source sync
PiperOrigin-RevId: 223530503
2018-12-02 11:50:39 -08:00
Peter Hawkins
6361b784a8 source sync
PiperOrigin-RevId: 222456068
2018-11-21 20:22:56 -08:00
Matthew Johnson
25fb9b421d source sync
PiperOrigin-RevId: 222170151
2018-11-21 20:22:33 -08:00
Roy Frostig
a3619ca89d source sync
PiperOrigin-RevId: 222153576
2018-11-21 20:22:30 -08:00
Matthew Johnson
50038c07c8 fix build file issues 2018-11-19 20:18:31 -08:00
Roy Frostig
99f98f8e8c source sync 2018-11-19 13:50:57 -08:00
Roy Frostig
da2d53ad33 source sync 2018-11-19 13:29:47 -08:00
Matthew Johnson
9ae0f3a610 split BUILD file, move up license files 2018-11-18 15:43:09 -08:00