diff --git a/tests/BUILD b/tests/BUILD index cd18370fc..59b93f316 100644 --- a/tests/BUILD +++ b/tests/BUILD @@ -816,14 +816,18 @@ jax_multiplatform_test( jax_multiplatform_test( name = "pytorch_interoperability_test", srcs = ["pytorch_interoperability_test.py"], + backend_tags = { + "gpu": [ + "noasan", # TODO(b/392599624): torch fails to build. + ], + }, enable_backends = [ "cpu", "gpu", ], tags = [ + "nomsan", # TODO(b/355237462): msan false-positives in torch? "not_build:arm", - # TODO(b/355237462): Re-enable once MSAN issue is addressed. - "nomsan", ], deps = py_deps("torch"), )