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

Disable some slow tests under asan.

Browse Source 
PiperOrigin-RevId: 509828659
This commit is contained in:
Peter Hawkins 2023-02-15 07:34:54 -08:00 committed by jax authors
parent 3bd6ca014c
commit 69b8a03400
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/BUILD
View File

@ -572,6 +572,11 @@ jax_test(
jax_test(
name = "pmap_test",
srcs = ["pmap_test.py"],
backend_tags = {
"tpu": [
"noasan", # Times out under asan.
],
},
pjrt_c_api_bypass = True,
shard_count = {
"cpu": 30,
@ -771,6 +776,11 @@ jax_test(
name = "sparsify_test",
srcs = ["sparsify_test.py"],
args = ["--jax_bcoo_cusparse_lowering=true"],
backend_tags = {
"tpu": [
"noasan", # Times out under asan.
],
},
shard_count = {
"cpu": 5,
"gpu": 20,