Mark mlir/XLA tests as passing with Sanitizer except gpu_ops.mlir
This is possible after 413001036 which broke the dependency on XLA GPU
which initializes Cuda and the GPU.
PiperOrigin-RevId: 413058845
Change-Id: Ic53f951c05da8f444c072d67b2b9877948b92eb0
diff --git a/tensorflow/compiler/mlir/xla/tests/BUILD b/tensorflow/compiler/mlir/xla/tests/BUILD
index 48d4baa..bae69f1 100644
--- a/tensorflow/compiler/mlir/xla/tests/BUILD
+++ b/tensorflow/compiler/mlir/xla/tests/BUILD
@@ -1,60 +1,12 @@
load("//tensorflow:tensorflow.bzl", "filegroup")
load("//tensorflow/compiler/mlir:glob_lit_test.bzl", "glob_lit_tests")
load("//tensorflow:tensorflow.bzl", "tf_cc_test")
-load(
- "//tensorflow/core/platform:build_config_root.bzl",
- "tf_cuda_tests_tags",
-)
package(licenses = ["notice"])
glob_lit_tests(
data = [":test_utilities"],
driver = "@llvm-project//mlir:run_lit.sh",
- tags_override = {
- "adjust-layout.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-BatchMatMulV2.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-binary-elementwise.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-collective.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-communication.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-control-flow.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-full-conversion.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-include-tf2xla-fallback.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-no-tf2xla-fallback.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-prefer-tf2xla.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-types.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "legalize-tf-with-tf2xla.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- "prepare-for-export.mlir": tf_cuda_tests_tags() + [
- "nomsan", # TODO(b/181135145)
- ],
- },
test_file_exts = [
"mlir",
"hlotxt",
diff --git a/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/BUILD b/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/BUILD
index e0d3757..0b5c571 100644
--- a/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/BUILD
+++ b/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/BUILD
@@ -17,18 +17,7 @@
"noubsan",
"no_cuda_asan",
"no_oss",
- ], # b/171751580
- "ops.mlir": tf_cuda_tests_tags() + [
- "nomsan",
- "no_cuda_asan",
- "no_oss",
- ], # b/191025174, b/192521861
- "passthrough.mlir": tf_cuda_tests_tags() + [
- "nomsan",
- "no_cuda_asan",
- "no_cuda_asan",
- "no_oss",
- ], # b/191025174, b/192521861
+ ],
},
test_file_exts = [
"mlir",
