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

[JAX] Disable msan and asan for the profiler test running on nvidia gpu

Browse Source 
PiperOrigin-RevId: 733380848
This commit is contained in:
jax authors 2025-03-04 10:32:32 -08:00
parent 7d0aab5a98
commit 8af6f70fe0
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/BUILD
View File

@ -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",