From 8af6f70fe0b3f375f41323cc60aadd347f1d2209 Mon Sep 17 00:00:00 2001
From: jax authors <google-ml-automation@google.com>
Date: Tue, 4 Mar 2025 10:32:32 -0800
Subject: [PATCH] [JAX] Disable msan and asan for the profiler test running on
 nvidia gpu

PiperOrigin-RevId: 733380848
---
 tests/BUILD | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/BUILD b/tests/BUILD
index d5d6c7be5..0ffa68ed8 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -838,6 +838,13 @@ jax_multiplatform_test(
 jax_multiplatform_test(
     name = "profiler_test",
     srcs = ["profiler_test.py"],
+    backend_tags = {
+        "gpu": [
+            # disable suspicious leaking in cupti/cuda,
+            # TODO: remove this once b/372714955 is resolved.
+            "noasan",
+        ],
+    },
     enable_backends = [
         "cpu",
         "gpu",