mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/tests
History
Matthew Johnson 71f5f9972c skip checks in big randomness test
2020-09-23 20:15:32 -07:00
..
benchmarks
Cleanup: remove unused imports in tests (#3276)
2020-06-01 11:49:35 -07:00
notebooks
Add Colab test notebooks for CPU, GPU, and TPU (#3000)
2020-05-08 11:11:42 -07:00
third_party/scipy
BFGS algorithm (#3101)
2020-07-29 14:14:40 -07:00
api_test.py
only test if omnistaging is enabled
2020-09-21 19:33:14 -07:00
api_util_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
array_interoperability_test.py
Cleanup: define type lists in test_util & use in several test files. (#3616)
2020-07-07 17:01:38 -07:00
batching_test.py
Fix code quality issues (#4302)
2020-09-17 09:21:18 -07:00
callback_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
core_test.py
rename and simplify TypedJaxpr -> ClosedJaxpr (#4328)
2020-09-18 10:07:13 -07:00
debug_nans_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
doubledouble_test.py
Remove some non-inclusive language (#3710)
2020-07-10 09:29:06 -07:00
dtypes_test.py
Remove type restrictions (#4017)
2020-08-10 20:48:20 -07:00
errors_test.py
unrevert #3674 (revert #3791)
2020-08-17 18:13:58 -07:00
fft_test.py
hfft and ihfft implementation (#3664)
2020-07-10 10:34:59 -07:00
generated_fun_test.py
Fix flaky generated_fun_test.py test (#3885)
2020-07-28 07:27:07 -07:00
gmap_test.py
Add support for binding axis_name in gmap
2020-08-28 14:42:01 +02:00
host_callback_test.py
Simplify the interface for host_callback.id_tap (#4101)
2020-09-14 12:47:28 +03:00
image_test.py
Updating image_test
2020-09-21 16:21:48 -07:00
infeed_test.py
omnistaging, under a flag and disabled by default (#3370)
2020-07-30 12:59:36 -07:00
jax_jit_test.py
Add more features to the C++ jax.jit. (#4169)
2020-09-01 10:34:47 +03:00
jax_to_hlo_test.py
fix missing imports
2020-06-29 11:08:57 -07:00
jaxpr_util_test.py
fix jaxpr util test under enable_x64
2020-08-19 08:28:56 -07:00
jet_test.py
cumulative jet rules (#4000)
2020-08-11 07:09:54 -07:00
lax_autodiff_test.py
Cleanup: avoid jnp.prod & np.prod on array shapes (#4086)
2020-08-18 10:17:38 -07:00
lax_control_flow_test.py
rename and simplify TypedJaxpr -> ClosedJaxpr (#4328)
2020-09-18 10:07:13 -07:00
lax_numpy_einsum_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
lax_numpy_indexing_test.py
Fixing weird behavior in segment_sum when num_segments is None (#4034)
2020-09-11 13:51:42 -04:00
lax_numpy_test.py
Merge pull request #4366 from jakevdp:diff-empty
2020-09-21 16:28:30 -07:00
lax_numpy_vectorize_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
lax_scipy_sparse_test.py
Refine argument validation inside jax.scipy.sparse.linalg.cg (#3630)
2020-07-06 09:24:44 -07:00
lax_scipy_test.py
Add Bessel functions in jax.numpy & jax.scipy.special (#4007)
2020-08-10 10:10:59 -07:00
lax_test.py
Fix shape checking rule for conv_general_dilated. (#4318)
2020-09-17 10:37:40 -07:00
lax_vmap_test.py
Add support for base dilation and window dilation to reduce window op… (#3803)
2020-07-20 17:27:24 -04:00
linalg_test.py
Enable tests for complex QR decomposition on TPU.
2020-09-23 09:08:17 -04:00
loops_test.py
unrevert #3674 (revert #3791)
2020-08-17 18:13:58 -07:00
masking_test.py
Support non-fragmenting mask of reshape (#4264)
2020-09-16 23:58:32 -07:00
metadata_test.py
omnistaging, under a flag and disabled by default (#3370)
2020-07-30 12:59:36 -07:00
multi_device_test.py
omnistaging, under a flag and disabled by default (#3370)
2020-07-30 12:59:36 -07:00
multibackend_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
nn_test.py
omnistaging on by default (#4038)
2020-09-15 08:06:46 -07:00
ode_test.py
Fix scan carry types in gradient of complex ODE (#4130)
2020-08-24 13:50:44 -07:00
optimizers_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
optix_test.py
fix missing imports
2020-06-29 11:08:57 -07:00
pmap_test.py
Fix axis_index inside nested pmaps
2020-09-22 16:41:46 +00:00
polynomial_test.py
Fix polynomial tests (#3933)
2020-08-01 08:37:31 -07:00
profiler_test.py
fix missing imports
2020-06-29 11:08:57 -07:00
random_test.py
skip checks in big randomness test
2020-09-23 20:15:32 -07:00
scipy_ndimage_test.py
Cleanup: avoid jnp.prod & np.prod on array shapes (#4086)
2020-08-18 10:17:38 -07:00
scipy_optimize_test.py
Fixes to test_scipy_optimize.py for Google internal tests (#3902)
2020-07-29 15:31:02 -07:00
scipy_signal_test.py
Cleanup: convert uses of 'import numpy as onp' in tests (#3756)
2020-07-14 13:03:24 -07:00
scipy_stats_test.py
Cleanup: convert uses of 'import numpy as onp' in tests (#3756)
2020-07-14 13:03:24 -07:00
sharded_jit_test.py
Avoid calling jnp.sum() on list (#4163)
2020-08-28 09:07:30 -07:00
stax_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
tree_util_tests.py
Compare treedefs by num_leaves not traversal_ in tree_transpose. (#3659)
2020-07-30 13:31:17 -04:00
util_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
vectorize_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
xla_bridge_test.py
fix missing imports
2020-06-29 11:08:57 -07:00