Sharad Vikram
6274b9ed39
Enable Python callbacks on TFRT TPU backend
...
PiperOrigin-RevId: 459415455
2022-07-06 20:52:50 -07:00
Peter Hawkins
95e79332c0
Add JAX_TEST_TARGETS and JAX_EXCLUDE_TEST_TARGETS environment variables to assist with skipping tests under Bazel.
...
Add "multiaccelerator" test tags to mark tests that would meaningfully run with more than one accelerator (e.g., GPU).
PiperOrigin-RevId: 459320212
2022-07-06 12:51:43 -07:00
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
