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
rocm_jax/tests
History
Jake VanderPlas 4c0d61a143 Add jtu.strict_promotion_if_dtypes_match utility
2022-06-16 13:59:53 -07:00
..
benchmarks
…
filecheck
[MHLO] Add CHLO lowering for top_k
2022-05-09 15:45:04 -07:00
notebooks
Update colab CPU test notebook to be more robust
2021-07-13 09:30:56 -07:00
third_party/scipy
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
ann_test.py
[JAX] Fix batch logic for approx_min/max_k
2022-04-08 13:50:36 -07:00
api_test.py
Merge pull request #11103 from jakevdp:x64-api-test
2022-06-15 17:22:32 -07: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
[ROCm] Enable XLA python callback on ROCm.
2022-06-03 17:52:34 +00:00
array_test.py
Add __array__ (for device_get), _npy_value, block_until_ready, delete and _check_if_deleted to Array.
2022-06-13 18:08:31 -07:00
batching_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
callback_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
checkify_test.py
[x64] deprecate unsafe type casting in scatter-update operations
2022-06-09 15:21:49 -07:00
compilation_cache_test.py
Run compilation_cache_test.py using TFRT
2022-02-04 10:17:59 -08:00
core_test.py
add core.closed_call_p
2022-05-14 14:06:30 -07:00
custom_linear_solve_test.py
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
custom_object_test.py
add support for lambda-bound dynamic shape output (iree only)
2022-05-18 21:57:30 -07:00
custom_root_test.py
Relax tolerances slightly for MKL.
2022-02-25 22:02:55 +00:00
debug_nans_test.py
feat: validate jit args
2022-05-18 21:54:47 +01:00
debugging_primitives_test.py
Enable debug print in xmap
2022-06-07 14:05:46 -07:00
distributed_test.py
Unbreak jax.distributed initialization.
2022-06-07 11:10:17 -07:00
dtypes_test.py
dtypes_test: run tests with jax_numpy_type_promotion=strict
2022-05-26 16:47:06 -07:00
errors_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
fft_test.py
[x64] make fft functionality compatible with strict dtype promotion
2022-06-15 10:10:44 -07:00
generated_fun_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
gfile_cache_test.py
Using etils(gfile) to support gcs buckets and file system for persistent compilation caching
2022-06-10 00:17:13 +00:00
global_device_array_test.py
Adding jax.Array to jax.experimental. Its pretty much the same as GDA (without the performance optimization for now).
2022-06-06 17:32:00 -07: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
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01: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
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
infeed_test.py
JaxTestCase: set jax_numpy_rank_promotion='raise' by default
2022-02-14 09:22:05 -08:00
jax_jit_test.py
feat: validate jit args
2022-05-18 21:54:47 +01:00
jax_to_ir_test.py
Fix name stack bugs
2022-04-19 11:14:41 -07:00
jaxpr_effects_test.py
Enable debug print in xmap
2022-06-07 14:05:46 -07:00
jaxpr_util_test.py
Increase minimum jaxlib version to 0.3.0.
2022-03-04 10:33:03 -05:00
jet_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
lax_autodiff_test.py
Fix autodiff of pad() of scalar inputs.
2022-06-01 15:42:56 -04:00
lax_control_flow_test.py
Add stateful for loop primitives (#10982)
2022-06-15 15:55:38 -07:00
lax_numpy_einsum_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
lax_numpy_indexing_test.py
[x64] deprecate unsafe type casting in scatter-update operations
2022-06-09 15:21:49 -07:00
lax_numpy_test.py
Add jtu.strict_promotion_if_dtypes_match utility
2022-06-16 13:59:53 -07:00
lax_numpy_vectorize_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
lax_scipy_sparse_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
lax_scipy_test.py
[x64] make lax_scipy_test.py compatible with strict dtype promotion
2022-06-14 10:02:45 -07:00
lax_test.py
Merge pull request #11114 from jakevdp:x64-lax-test
2022-06-15 17:44:23 -07:00
lax_vmap_test.py
Fix batching rule for convolution for batch dimensions of size 0.
2022-06-01 14:18:16 -04:00
linalg_test.py
[TPU] Switch the default eigendecomposition implementation on TPU to use QDWH-eig.
2022-05-27 13:50:37 -07:00
masking_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
mesh_utils_test.py
[mesh_utils] Avoid relying on process-tiled device order
2022-06-06 13:30:56 -07:00
metadata_test.py
Fix name stack bugs
2022-04-19 11:14:41 -07:00
multi_device_test.py
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
multibackend_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
name_stack_test.py
Only wrap one level of the name-stack with transform names.
2022-06-10 03:50:26 -07:00
nn_test.py
Rename jax.nn.normalize to standardize. Add normalize alias with DeprecationWarning.
2022-03-23 20:55:22 +00:00
ode_test.py
[XLA:CPU] Relax test tolerances for tests using XLA:CPU.
2022-02-28 09:26:54 -08:00
optimizers_test.py
Testing: avoid hard-coding random seeds
2021-12-10 10:32:09 -08:00
pickle_test.py
DeviceArray: Improve support for copy, deepcopy, and pickle
2022-05-19 12:00:58 -07:00
pjit_test.py
Add __array__ (for device_get), _npy_value, block_until_ready, delete and _check_if_deleted to Array.
2022-06-13 18:08:31 -07:00
pmap_test.py
pmap_lib should use the same cache key for threadlocal context as python jax.
2022-06-10 21:39:15 -07:00
polynomial_test.py
Testing: avoid hard-coding random seeds
2021-12-10 10:32:09 -08:00
profiler_test.py
Add collect_profile script
2022-06-03 17:56:17 -07:00
qdwh_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
random_test.py
Added random.generalized_normal and random.ball.
2022-06-03 15:11:29 -04:00
remote_transfer_test.py
Increase minimum jaxlib version to 0.3.7.
2022-04-18 08:09:50 -04:00
scipy_fft_test.py
Update JAX to use new math libraries in ROCm-5.0.
2022-03-01 20:02:15 +00:00
scipy_interpolate_test.py
Merge pull request #8625 from Edenhofer:regular_grid_interpolator
2022-01-24 12:02:34 -08:00
scipy_ndimage_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
scipy_optimize_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
scipy_signal_test.py
Disabled one and enabled several unit tests for ROCm.
2022-05-10 19:47:26 +00:00
scipy_stats_test.py
Add jtu.strict_promotion_if_dtypes_match utility
2022-06-16 13:59:53 -07:00
sharded_jit_test.py
Remove unnecessary __future__ imports
2022-05-12 11:57:21 -07:00
sparse_test.py
Add jtu.strict_promotion_if_dtypes_match utility
2022-06-16 13:59:53 -07:00
sparsify_test.py
[x64] make sparse tests compatible with strict dtype promotion
2022-06-14 12:45:26 -07:00
stack_test.py
Make QDWH-eig implementation jit-table.
2022-05-17 18:12:48 -07:00
stax_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
svd_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
transfer_guard_test.py
Bump minimum jaxlib version to 0.3.2 and remove transfer guard compatibility code
2022-04-11 15:33:27 +00:00
tree_util_test.py
[x64] make tree_util_test compatible with strict dtype promotion
2022-06-14 15:14:44 -07:00
util_test.py
Fix race condition for weakref destructor by catching rare exceptions.
2022-04-01 12:04:36 -07:00
version_test.py
Introduce a new jax/jaxlib versioning scheme.
2022-02-07 17:59:42 -05:00
x64_context_test.py
feat: validate jit args
2022-05-18 21:54:47 +01:00
xla_bridge_test.py
Only register backend factories once when jax_platforms config value is set.
2022-05-20 09:46:37 -07:00
xla_interpreter_test.py
Split jax.interpreters.xla up into three pieces:
2021-11-22 08:22:43 -08:00
xmap_test.py
Add Array support to xmap. Just using the GDA path.
2022-06-13 07:31:03 -07:00