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

Do not run nn_test under asan -- it times out

Browse Source 
PiperOrigin-RevId: 660377176
This commit is contained in:
Sergei Lebedev 2024-08-07 07:13:25 -07:00 committed by jax authors
parent 3e5e947542
commit 3a1567f57a
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/BUILD
View File

@ -688,6 +688,11 @@ jax_test(
jax_test(
name = "nn_test",
srcs = ["nn_test.py"],
backend_tags = {
"gpu": [
"noasan", # Times out under asan.
],
},
shard_count = {
"cpu": 10,
"tpu": 10,