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 asan for torch interoperability test on CPU.

Browse Source 
This is timing out at build time on CI.

PiperOrigin-RevId: 609350782
This commit is contained in:
Peter Hawkins 2024-02-22 06:28:32 -08:00 committed by jax authors
parent 7c0a1214bb
commit 4ff84d04d9
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/BUILD
View File

@ -730,12 +730,13 @@ jax_test(
jax_test(
name = "pytorch_interoperability_test",
srcs = ["pytorch_interoperability_test.py"],
backend_tags = {
# PyTorch leaks dlpack metadata https://github.com/pytorch/pytorch/issues/117058
"gpu": ["noasan"],
},
disable_backends = ["tpu"],
tags = ["not_build:arm"],
tags = [
# PyTorch leaks dlpack metadata https://github.com/pytorch/pytorch/issues/117058, and
# compilation times out on CPU.
"noasan",
"not_build:arm",
],
deps = py_deps("torch"),
)