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:
parent
e7e5bc9440
commit
cf9b201689
@ -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 = [
|
||||
|
||||
@ -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()
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user