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

Disable sanitizer builds of shape_poly_test.

Browse Source 
These take a very long time and sometimes timeout so it's probably not worth running them in CI.

PiperOrigin-RevId: 587768399
This commit is contained in:
Peter Hawkins 2023-12-04 10:28:28 -08:00 committed by jax authors
parent 54e3b7611a
commit 5b97960d31
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/BUILD
View File

@ -1307,13 +1307,6 @@ jax_test(
jax_test(
name = "shape_poly_test",
srcs = ["shape_poly_test.py"],
backend_tags = {
"tpu": [
"noasan",
"nomsan",
"notsan",
], # Times out
},
disable_configs = [
"gpu_a100", # TODO(b/269593297): matmul precision issues
],
@ -1326,6 +1319,11 @@ jax_test(
"gpu": 4,
"tpu": 4,
},
tags = [
"noasan", # Times out
"nomsan", # Times out
"notsan", # Times out
],
deps = [
"//jax:internal_test_harnesses",
"//jax/experimental/export",