From cf9b201689025e71e2d637f4b5b49732c64f5815 Mon Sep 17 00:00:00 2001 From: Jay Shi Date: Mon, 14 Sep 2020 15:02:43 -0700 Subject: [PATCH] Enable MLIR bridge for tpu_outside_compilation_test but disable one test. A single test target fails with MLIR bridge. Enables bridge for all other cases to prevent regressions. PiperOrigin-RevId: 331636353 Change-Id: I5e17c9bf0afb058b86f291916d69271772e0179c --- tensorflow/python/tpu/BUILD | 1 - tensorflow/python/tpu/tpu_outside_compilation_test.py | 4 ---- 2 files changed, 5 deletions(-) diff --git a/tensorflow/python/tpu/BUILD b/tensorflow/python/tpu/BUILD index 01f0e95c0b9..9cd75d1bed7 100644 --- a/tensorflow/python/tpu/BUILD +++ b/tensorflow/python/tpu/BUILD @@ -551,7 +551,6 @@ tpu_py_test( "tpu_outside_compilation_test.py", ], disable_experimental = True, - disable_mlir_bridge = False, python_version = "PY3", tags = ["no_oss"], deps = [ diff --git a/tensorflow/python/tpu/tpu_outside_compilation_test.py b/tensorflow/python/tpu/tpu_outside_compilation_test.py index d09a2f70088..5ae618640b2 100644 --- a/tensorflow/python/tpu/tpu_outside_compilation_test.py +++ b/tensorflow/python/tpu/tpu_outside_compilation_test.py @@ -32,7 +32,6 @@ from tensorflow.python.eager import remote from tensorflow.python.eager import test from tensorflow.python.framework import config from tensorflow.python.framework import constant_op -from tensorflow.python.framework import test_util from tensorflow.python.lib.io import tf_record from tensorflow.python.ops import array_ops from tensorflow.python.ops import control_flow_ops @@ -555,9 +554,6 @@ class OutsideCompilationOnUnsupportedOpTest(test.TestCase, self.assertEqual(events[1].summary.value[0].tag, "cond/x") self.assertEqual(events[1].summary.value[0].simple_value, 3.0) - @test_util.disable_mlir_bridge( - "TODO(b/168493455): Reenable this test one deadlock resolved." - ) def testAutoOutsideCompilationWithFunctionalNodes(self): strategy = get_tpu_strategy()