mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/tests
History
Peter Hawkins 417de0d351
Add jit to jax.image.resize (#3714) 
* Add image/ directory to Bazel build.

* Use a jit on jax.image.resize to reduce compilation time.

Relax bfloat16 test tolerance.
2020-07-10 10:32:13 -04: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
api_test.py
improve vmap axis spec structure mismatch errors (#3619)
2020-06-30 22:19:16 -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
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
callback_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
core_test.py
Add ability to specify individual test targets via a regex (#3549)
2020-06-29 14:16:51 -07:00
debug_nans_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
doubledouble_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
dtypes_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
fft_test.py
Add _NOT_IMPLEMENTED attribute to jax.numpy (fixes #3689) (#3698)
2020-07-09 16:31:08 -07:00
generated_fun_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
host_callback_test.py
Added concurrent id_tap tests, disabled for GPU (#3690)
2020-07-08 16:08:54 +03:00
image_test.py
Add jit to jax.image.resize (#3714)
2020-07-10 10:32:13 -04:00
infeed_test.py
Refactored host_callback to use the C++ runtime. (#3644)
2020-07-04 18:12:58 +03:00
jax_to_hlo_test.py
fix missing imports
2020-06-29 11:08:57 -07:00
jet_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
lax_autodiff_test.py
Cleanup: define type lists in test_util & use in several test files. (#3616)
2020-07-07 17:01:38 -07:00
lax_control_flow_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -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
Cleanup: use test_util dtypes where possible (#3695)
2020-07-08 13:21:48 -07:00
lax_numpy_test.py
Add _NOT_IMPLEMENTED attribute to jax.numpy (fixes #3689) (#3698)
2020-07-09 16:31:08 -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
Cleanup: use test_util dtypes where possible (#3695)
2020-07-08 13:21:48 -07:00
lax_test.py
Generalize lax.sort to support lexicographic sorts. (#3709)
2020-07-09 20:05:19 -07:00
lax_vmap_test.py
Cleanup: define type lists in test_util & use in several test files. (#3616)
2020-07-07 17:01:38 -07:00
linalg_test.py
Add _NOT_IMPLEMENTED attribute to jax.numpy (fixes #3689) (#3698)
2020-07-09 16:31:08 -07:00
loops_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
masking_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
metadata_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
multi_device_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
multibackend_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
nn_test.py
improve concreteness error message for nn.one_hot (#3656)
2020-07-03 20:54:25 -07:00
ode_test.py
skip some ode tests on gpu for speed (#3629)
2020-07-01 11:26: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
parallel_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
pmap_test.py
fix bad pmap tests from #3675 (#3685)
2020-07-07 14:48:54 -07:00
polynomial_test.py
Add _NOT_IMPLEMENTED attribute to jax.numpy (fixes #3689) (#3698)
2020-07-09 16:31:08 -07:00
profiler_test.py
fix missing imports
2020-06-29 11:08:57 -07:00
random_test.py
Cleanup: define type lists in test_util & use in several test files. (#3616)
2020-07-07 17:01:38 -07:00
scipy_ndimage_test.py
Cleanup: use test_util dtypes where possible (#3695)
2020-07-08 13:21:48 -07:00
scipy_signal_test.py
Cleanup: define type lists in test_util & use in several test files. (#3616)
2020-07-07 17:01:38 -07:00
scipy_stats_test.py
Cleanup: define type lists in test_util & use in several test files. (#3616)
2020-07-07 17:01:38 -07:00
sharded_jit_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
stax_test.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07:00
tree_util_tests.py
Add ability to specify individual test targets
2020-06-29 11:08:57 -07: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