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 tests that time out or OOM under ASAN.

Browse Source 
PiperOrigin-RevId: 598543036
This commit is contained in:
Peter Hawkins 2024-01-15 01:40:05 -08:00 committed by jax authors
parent 66ee75ad19
commit 912a5ef771
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/BUILD
View File

@ -337,6 +337,7 @@ jax_test(
"tpu": 10,
"iree": 10,
},
tags = ["noasan"], # Linking TF causes a linker OOM.
deps = py_deps("pil") + py_deps("tensorflow_core"),
)
@ -1150,6 +1151,7 @@ jax_test(
jax_test(
name = "host_callback_to_tf_test",
srcs = ["host_callback_to_tf_test.py"],
tags = ["noasan"], # Linking TF causes a linker OOM.
deps = [
"//jax:experimental_host_callback",
"//jax:ode",

1
tests/pallas/BUILD
View File

@ -93,6 +93,7 @@ jax_test(
],
shard_count = 18,
tags = [
"noasan", # Times out.
"nomsan", # Times out.
"notsan", # Times out.
],