mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

random_test: Split into two so that each target is small enough to fit within a medium timeout.

Browse Source 
PiperOrigin-RevId: 571146766
This commit is contained in:
jax authors 2023-10-05 15:28:24 -07:00
parent 15126504a7
commit 8f911e1512
3 changed files with 1412 additions and 1310 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tests/BUILD
View File

@ -704,10 +704,28 @@ jax_test(
"notsan", # Times out "notsan", # Times out
], ],
}, },
# Use fewer cases to prevent timeouts. shard_count = {
backend_variant_args = { "cpu": 30,
"cpu": ["--jax_num_generated_cases=40"], "gpu": 30,
"cpu_x32": ["--jax_num_generated_cases=40"], "tpu": 40,
"iree": 30,
},
tags = ["noasan"], # Times out
)
jax_test(
name = "random_lax_test",
srcs = ["random_lax_test.py"],
backend_tags = {
"cpu": [
"notsan", # Times out
"nomsan", # Times out
],
"tpu": [
"optonly",
"nomsan", # Times out
"notsan", # Times out
],
}, },
shard_count = { shard_count = {
"cpu": 30, "cpu": 30,

1389
tests/random_lax_test.py Normal file
View File

File diff suppressed because it is too large Load Diff

1307
tests/random_test.py
View File

File diff suppressed because it is too large Load Diff