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
jax authors d7ec244c0c Merge pull request #11160 from jakevdp:x64-xmap-test 
PiperOrigin-RevId: 456274863
2022-06-21 09:13:38 -07:00
..
benchmarks
Cleanup: remove unused imports in tests (#3276)
2020-06-01 11:49:35 -07:00
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
[new-remat] add _scan_partial_eval_custom rule for new remat
2022-06-17 23:15:14 -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 __repr__ to Array. It works exactly as it does for DA and SDA when it is fully addressable. Otherwise it works like GDA.
2022-06-17 13:16:49 -07:00
batching_test.py
[x64] make batching_test compatible with strict dtype promotion
2022-06-17 16:08:54 -07:00
callback_test.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
checkify_test.py
Checkify: support discharging checks from control-flow through effects.
2022-06-20 18:28:03 +01:00
compilation_cache_test.py
Run compilation_cache_test.py using TFRT
2022-02-04 10:17:59 -08:00
core_test.py
Merge pull request #11151 from mattjj:input-fwd-residual-optimization
2022-06-17 15:43:44 -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
[x64] make host_callback_test compatible with strict dtype promotion
2022-06-16 14:37:44 -07: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
[x64] make jaxpr_effects_test pass with strict dtype promotion
2022-06-16 14:28:16 -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
[new-remat] add _scan_partial_eval_custom rule for new remat
2022-06-17 23:15:14 -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] make lax_numpy_indexing_test pass with strict dtype promotion
2022-06-16 14:00:11 -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
[x64] make lax_numpy_vectorize_test work with strict dtype promotion
2022-06-16 14:06:24 -07:00
lax_scipy_sparse_test.py
[x64] make jax.scipy.sparse.linalg compatible with strict dtype promotion
2022-06-17 14:04:05 -07: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
[x64] make vmap_test compatible with strict dtype promotion
2022-06-17 15:55:08 -07:00
linalg_test.py
[x64] make linalg functions & tests compatible with strict dtype promotion
2022-06-16 10:32:20 -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
[x64] make ode_test pass with strict dtype promotion
2022-06-17 15:21:10 -07: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
[x64] make pjit_test compatible with strict dtype promotion
2022-06-17 16:25:36 -07:00
pmap_test.py
[new-remat] add _scan_partial_eval_custom rule for new remat
2022-06-17 23:15:14 -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
[x64] make scipy_ndimage_test compatible with strict dtype promotion
2022-06-16 15:45:15 -07:00
scipy_optimize_test.py
BUG: fix incorrect order of operations in line search
2022-06-16 15:21:24 -07: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
[x64] make sharded_jit_test compatible with strict dtype promotion
2022-06-17 16:42:45 -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
[x64] make linalg functions & tests compatible with strict dtype promotion
2022-06-16 10:32:20 -07: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
[JAX] Drop the private _process_pytree method from tree_util.
2022-06-17 13:50:50 -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
[x64] make xmap_test compatible with strict dtype promotion
2022-06-17 16:35:42 -07:00