Enable MLIR bridge for all tests in compiler/ that already work with it.
PiperOrigin-RevId: 381533826
Change-Id: I09e471aea1d4423e979f0a9130079a784d890341
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD
index 6b09687..2367d19 100644
--- a/tensorflow/compiler/tests/BUILD
+++ b/tensorflow/compiler/tests/BUILD
@@ -337,7 +337,7 @@
name = "self_adjoint_eig_op_test",
size = "medium",
srcs = ["self_adjoint_eig_op_test.py"],
- enable_mlir_bridge = False,
+ enable_mlir_bridge = True,
python_version = "PY3",
tags = [
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip
@@ -403,7 +403,7 @@
size = "small",
timeout = "moderate",
srcs = ["matrix_inverse_op_test.py"],
- enable_mlir_bridge = False,
+ enable_mlir_bridge = True,
python_version = "PY3",
tags = [
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip
@@ -426,7 +426,7 @@
size = "small",
timeout = "moderate",
srcs = ["matrix_solve_op_test.py"],
- enable_mlir_bridge = False,
+ enable_mlir_bridge = True,
python_version = "PY3",
tags = [
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip