[TF2XLA] Re-enable tf.function + MLIR bridge tests, disable failing test
PiperOrigin-RevId: 324088917
Change-Id: I52c02d8a274bb5f0d1811e8368bb351da7d5952f
diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD
index ebc9d67..8da45b0 100644
--- a/tensorflow/python/eager/BUILD
+++ b/tensorflow/python/eager/BUILD
@@ -850,7 +850,7 @@
tf_xla_py_test(
name = "def_function_xla_jit_test",
srcs = ["def_function_xla_jit_test.py"],
- enable_mlir_bridge = False, # TODO(b/162521846) change back to True
+ enable_mlir_bridge = True,
enabled_backends = [
# TODO(b/162438052): Enable the test on TPU.
"cpu",
diff --git a/tensorflow/python/eager/def_function_xla_jit_test.py b/tensorflow/python/eager/def_function_xla_jit_test.py
index 8109e79..1098207 100644
--- a/tensorflow/python/eager/def_function_xla_jit_test.py
+++ b/tensorflow/python/eager/def_function_xla_jit_test.py
@@ -200,6 +200,8 @@
# XLA support is not yet enabled for TF ROCm
run_and_check(xla_func)
+ @test_util.disable_mlir_bridge('TODO(b/162521846): MLIR bridge fails'
+ ' msan, function library not found')
def testControlFlow(self):
with ops.device('device:{}:0'.format(self.device)):