Matthew Johnson
5d2fc7c05b
only test np.polyval on nonscalar array shapes
2018-12-30 18:07:50 -08:00
Roy Frostig
b2d6ce175a
add a placeholder comment for lbr tests
2018-12-28 13:51:32 -08:00
Matthew Johnson
0a9ee106b3
implement triangular solve lhs jvp (w/ @froystig)
2018-12-18 23:39:46 -08:00
Matthew Johnson
13b8e21a1c
squash conv grad bug introduced in 0d64aea
...
(loudly errored, didn't produce silently incorrect results!)
2018-12-14 18:40:50 -08:00
Matthew Johnson
6de5c8a698
add test-running instructions ( fixes #67 )
2018-12-14 16:48:08 -08:00
Matthew Johnson
bf5a30cdbd
fix warning
2018-12-13 11:55:11 -08:00
Matthew Johnson
54bceee9e1
make num_generated_cases also settable by env var
2018-12-13 08:56:40 -08:00
Matthew Johnson
0d64aea6bb
clean up conv dimension_numbers handling
2018-12-11 13:22:07 -05:00
Dougal Maclaurin
1350db2b79
Added higher-order differentiation checks in lax_test and fixed some bugs. Conv tests currently failing.
2018-12-11 13:22:07 -05:00
Peter Hawkins
38927153b1
Fix support for arrays with size-0 dimensions.
...
* Fix broadcasting rules to support size-0 dimensions.
* Add tests for size-0 dimensions. This required extending the test harness to support testing shapes that aren't necessarily broadcast compatible.
* Fix test utils to support size-0 dimensions.
2018-12-10 09:00:22 -05:00
Roy Frostig
db748a046f
py3 compatibility fix
2018-12-07 10:00:14 -08:00
Dougal Maclaurin
1627827ac6
Fixed a couple of bugs
2018-12-06 21:47:47 -05:00
Dougal Maclaurin
494c2dbae2
Merge branch 'master' into quickish-check
2018-12-06 18:58:26 -05:00
Dougal Maclaurin
8b88027df0
Number of test cases settable with command-line flag
2018-12-06 18:30:59 -05:00
Dougal Maclaurin
ebc6cd1e03
Step through sizes in test case generation
2018-12-06 18:02:43 -05:00
Dougal Maclaurin
3dbf41f3e6
Generated function tests working with bazel
2018-12-06 17:31:52 -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
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
Roy Frostig
642046e2c0
Bridge to the XRT backend.
...
PiperOrigin-RevId: 222763810
2018-11-27 16:51:06 -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
a30e858e59
populating source tree
2018-11-17 18:03:33 -08:00