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
This commit is contained in:
Jay Shi 2020-09-14 15:02:43 -07:00 committed by TensorFlower Gardener
parent e7e5bc9440
commit cf9b201689
2 changed files with 0 additions and 5 deletions

View File

@ -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 = [

View File

@ -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()