mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
17,366 Commits 506 Branches 263 Tags
Commit Graph

6382 Commits

Author SHA1 Message Date
Dougal Maclaurin
29113dd606 Made tests runnable with bazel 2018-12-06 17:00:47 -05:00
Peter Hawkins
c1b9eb19ea [JAX] Change semantics of dtype promotion to just call numpy.result_type. 
* Enable tests for numpy scalars in lax_numpy_test.py.
* Fix invalid promotion in random.py.
* Split tests for bitwise ops into their own test case and test mixed signedness.
* Add complex64 to the set of types supported by abstractify.
 2018-12-06 13:25:42 -05:00
Peter Hawkins
f2795cbdea [JAX] Add a NUMPY_SCALAR_SHAPE constant shape to test_utils.py to allow tests for both numpy scalars as distinct from 0D ndarrays. 
Fix mishandling of scalars when passed to np.reshape().

PiperOrigin-RevId: 224326107
 2018-12-06 06:47:58 -08:00
Dougal Maclaurin
727755178d Added license header 2018-12-05 15:55:01 -05:00
Dougal Maclaurin
2e4ff400fa Fixed bug in vjp with constant-zero tangent outputs 2018-12-05 15:55:01 -05:00
Dougal Maclaurin
2f44eba01d Made some subset of vjp/jvp inputs static in quickercheck. Exposing bugs. 2018-12-05 15:55:01 -05:00
Dougal Maclaurin
f1d7ea8972 Added reverse-mode checks 2018-12-05 15:55:01 -05:00
Dougal Maclaurin
6fcee12cec Added forward derivative checks 2018-12-05 15:55:01 -05:00
Dougal Maclaurin
5bb8f87e22 First pass at a quickcheck-style property-based tester to test invariants of jit/grad/vmap etc on random functions 2018-12-05 15:55:01 -05:00
Matthew Johnson
2597300c7f source sync 
PiperOrigin-RevId: 224157599
 2018-12-05 09:36:16 -08: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
Dougal Maclaurin
2df36f7510 Made a shim to handle configuration without having absl parse command-line flags. 
PiperOrigin-RevId: 223391288
 2018-11-29 13:44:54 -08:00
Matthew Johnson
1d2aaad6fe fix handling of symbolic zeros for a few special primitives 
PiperOrigin-RevId: 223264329
 2018-11-29 12:06:17 -08:00
Sam Schoenholz
326773808b source sync 
PiperOrigin-RevId: 222867729
 2018-11-27 16:51:15 -08:00
Roy Frostig
642046e2c0 Bridge to the XRT backend. 
PiperOrigin-RevId: 222763810
 2018-11-27 16:51:06 -08:00
Matthew Johnson
3b3490f406 source sync 
PiperOrigin-RevId: 222483357
 2018-11-21 20:32:33 -08:00
Matthew Johnson
ab53373665 source sync 
PiperOrigin-RevId: 222470141
 2018-11-21 20:32:33 -08:00
Peter Hawkins
6361b784a8 source sync 
PiperOrigin-RevId: 222456068
 2018-11-21 20:22:56 -08:00
Peter Hawkins
5e60639bc5 source sync 
PiperOrigin-RevId: 222452709
 2018-11-21 20:22:54 -08:00
Peter Hawkins
e180f08113 source sync 
PiperOrigin-RevId: 222451919
 2018-11-21 20:22:51 -08:00
Matthew Johnson
e5b76f4fde source sync 
PiperOrigin-RevId: 222340967
 2018-11-21 20:22:43 -08:00
Matthew Johnson
7f546b8c02 source sync 
PiperOrigin-RevId: 222175432
 2018-11-21 20:22:35 -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
5e9fcbb6e8 source sync 2018-11-19 21:31:39 -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
46c6a9170f sync updates 2018-11-19 07:47:59 -08:00
Matthew Johnson
9ae0f3a610 split BUILD file, move up license files 2018-11-18 15:43:09 -08:00
Matthew Johnson
489dd1c81a move jax tests and examples to repo root 2018-11-18 15:15:47 -08:00