From 4ff84d04d98defa4d3c72a3b40b3c136c7a1870d Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Thu, 22 Feb 2024 06:28:32 -0800 Subject: [PATCH] Disable asan for torch interoperability test on CPU. This is timing out at build time on CI. PiperOrigin-RevId: 609350782 --- tests/BUILD | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/BUILD b/tests/BUILD index 5c6b35e5a..1e64d98db 100644 --- a/tests/BUILD +++ b/tests/BUILD @@ -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"), )