Skip to content

Commit

Permalink
build_config_root: Disable tsan for GPU tests.
Browse files Browse the repository at this point in the history
CUDA doesn't support TSAN, and we believe ROC-m doesn't either.

PiperOrigin-RevId: 689838288
  • Loading branch information
pizzud authored and Google-ML-Automation committed Oct 25, 2024
1 parent 751cece commit ee3dbcf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion xla/tests/build_defs.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ def prepare_nvidia_gpu_backend_data(backends, disabled_backends, backend_tags, b
new_disabled_backends.extend(NVIDIA_GPU_BACKENDS)

new_backend_tags = {key: value for key, value in backend_tags.items() if key != "gpu"}
gpu_backend_tags = backend_tags.get("gpu", [])
gpu_backend_tags = backend_tags.get("gpu", tf_gpu_tests_tags())
for key in NVIDIA_GPU_BACKENDS:
new_backend_tags.setdefault(key, gpu_backend_tags[:])

Expand Down

0 comments on commit ee3dbcf

Please sign in to comment.