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

Bump shard count for experimental_rnn_test, which is timing out in CI when built under ASAN.

Browse Source 
PiperOrigin-RevId: 635850400
This commit is contained in:
Peter Hawkins 2024-05-21 10:24:29 -07:00 committed by jax authors
parent 5350bc960d
commit 483f924ea1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/BUILD
View File

@ -1354,7 +1354,7 @@ jax_test(
disable_configs = [
"gpu_a100", # Numerical precision problems.
],
shard_count = 8,
shard_count = 15,
deps = [
"//jax:rnn",
],