mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/tests
History
Peter Hawkins 4679f455f9 Change the default out-of-bounds behavior for jax.ops.segment_... to FILL_OR_DROP. 
This matches the documented behavior.

Fixes https://github.com/google/jax/issues/8634

PiperOrigin-RevId: 411635687
2021-11-22 13:32:58 -08:00
..
benchmarks
Cleanup: remove unused imports in tests (#3276)
2020-06-01 11:49:35 -07:00
notebooks
Update colab CPU test notebook to be more robust
2021-07-13 09:30:56 -07:00
third_party/scipy
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
ann_test.py
[JAX] Update JAX users in preparation for a change that makes iteration over a JAX array return JAX arrays, instead of NumPy arrays.
2021-10-28 16:49:37 -07:00
api_test.py
Split jax.interpreters.xla up into three pieces:
2021-11-22 08:22:43 -08:00
api_util_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
array_interoperability_test.py
update test of dlpack error message
2021-10-07 19:14:13 -07:00
batching_test.py
make vmap ppermute consistent with pmap/docstring
2021-11-18 14:02:49 -08:00
callback_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
compilation_cache_test.py
Include XLA_FLAGS in persistent compilation cache key.
2021-10-06 14:11:40 -07:00
core_test.py
Change jax.core.DropVar to be a non-singleton.
2021-10-18 15:02:54 -07:00
custom_object_test.py
Split jax.interpreters.xla up into three pieces:
2021-11-22 08:22:43 -08:00
debug_nans_test.py
Increase minimum jaxlib version to 0.1.74.
2021-11-18 15:06:58 -05:00
djax_test.py
[JAX] Fix pylint errors.
2021-10-04 17:54:46 -07:00
dtypes_test.py
Split jax.interpreters.xla up into three pieces:
2021-11-22 08:22:43 -08:00
errors_test.py
[JAX] Fix pylint errors.
2021-10-04 17:54:46 -07:00
fft_test.py
add fft normalisation
2021-10-20 22:15:35 +01:00
file_system_cache_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
generated_fun_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
gsda_test.py
Add local_data API for GSDA
2021-11-19 18:56:17 -08:00
heap_profiler_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
host_callback_test.py
generalize jaxpr simplification machinery
2021-11-19 09:00:59 -08:00
host_callback_to_tf_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
image_test.py
Added workaround for bug in XLA
2021-11-18 11:01:50 +02:00
infeed_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
jax_jit_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
jax_to_hlo_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
jaxpr_util_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
jet_test.py
[JAX] move example libraries from jax.experimental into jax.example_libraries
2021-10-19 17:30:45 -07:00
lax_autodiff_test.py
Fix incorrect gradient for base-dilated reduce window.
2021-11-22 09:34:10 -08:00
lax_control_flow_test.py
Split jax.interpreters.xla up into three pieces:
2021-11-22 08:22:43 -08:00
lax_numpy_einsum_test.py
[JAX] Fix pylint errors.
2021-10-04 17:54:46 -07:00
lax_numpy_indexing_test.py
Change the default out-of-bounds behavior for jax.ops.segment_... to FILL_OR_DROP.
2021-11-22 13:32:58 -08:00
lax_numpy_test.py
Split jax.interpreters.xla up into three pieces:
2021-11-22 08:22:43 -08:00
lax_numpy_vectorize_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
lax_scipy_sparse_test.py
Reenable a test that was disabled due to an (apparently fixed) LLVM bug.
2021-10-16 10:34:36 -07:00
lax_scipy_test.py
Skip some polar decomposition tests that fail on A100.
2021-11-22 11:18:22 -08:00
lax_test.py
Make lax.reduce_window variadic.
2021-11-22 13:21:37 -08:00
lax_vmap_test.py
Implement lax.reduce() batching rule case where batch dimensions differ between operands.
2021-11-03 09:42:01 -04:00
linalg_test.py
Increase minimum jaxlib version to 0.1.74.
2021-11-18 15:06:58 -05:00
loops_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
masking_test.py
[JAX] Fix pylint errors.
2021-10-04 17:54:46 -07:00
metadata_test.py
[JAX] Fix pylint errors.
2021-10-04 17:54:46 -07:00
multi_device_test.py
Split jax.interpreters.xla up into three pieces:
2021-11-22 08:22:43 -08:00
multibackend_test.py
Cleanup internal representation of XLA translation rules.
2021-10-16 07:53:24 -07:00
nn_test.py
fix whitespace
2021-10-27 22:06:06 -07:00
ode_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
optimizers_test.py
[JAX] move example libraries from jax.experimental into jax.example_libraries
2021-10-19 17:30:45 -07:00
pickle_test.py
Increase minimum jaxlib version to 0.1.74.
2021-11-18 15:06:58 -05:00
pjit_test.py
Adding support for a special value for in_axis_resources (pjit.FROM_GSDA) when GSDA is an input.
2021-11-19 14:48:11 -08:00
pmap_test.py
Split jax.interpreters.xla up into three pieces:
2021-11-22 08:22:43 -08:00
polynomial_test.py
[JAX] Fix pylint errors.
2021-10-04 17:54:46 -07:00
profiler_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
qdwh_test.py
Disable QDWH tests that fail on GPU and TPU.
2021-11-22 10:21:41 -08:00
random_test.py
Overload jnp ops which are polymorphic to an array's value and support PRNGKeys.
2021-11-16 23:00:32 +00:00
scipy_fft_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
scipy_ndimage_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
scipy_optimize_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
scipy_signal_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
scipy_stats_test.py
stats.multivariate_normal: support broadcasted inputs
2021-10-19 16:58:36 -07:00
sharded_jit_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
sparse_test.py
[sparse] spdot_general: implement many more cases
2021-11-18 14:32:58 -08:00
sparsify_test.py
[sparse]: add tracer-based implementation of sparsify
2021-11-04 13:02:46 -07:00
stax_test.py
[JAX] move example libraries from jax.experimental into jax.example_libraries
2021-10-19 17:30:45 -07:00
tree_util_test.py
Increase minimum jaxlib version to 0.1.74.
2021-11-18 15:06:58 -05:00
util_test.py
Move jax.test_util to jax._src.test_util.
2021-09-24 07:02:49 -07:00
x64_context_test.py
generalize jaxpr simplification machinery
2021-11-19 09:00:59 -08:00
xla_bridge_test.py
Remove xla_bridge.make_computation_builder().
2021-10-18 13:20:34 -04:00
xla_interpreter_test.py
Split jax.interpreters.xla up into three pieces:
2021-11-22 08:22:43 -08:00
xmap_test.py
Use shaped_abstractify to allow abstract arguments in .lower methods
2021-09-27 05:32:12 -07:00