experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enable MLIR bridge for the passing compiler tests

Browse Source 
PiperOrigin-RevId: 327094658
Change-Id: I1b7960b85ed68a0744cec7bbcee0190a5d0cbdb6
This commit is contained in:
Smit Hinsu 2020-08-17 14:05:25 -07:00 committed by TensorFlower Gardener
parent 589587081e
commit 30bd504acc
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/tests/BUILD
View File

@ -349,6 +349,7 @@ tf_xla_py_test(
size = "small", size = "small",
timeout = "moderate", timeout = "moderate",
srcs = ["searchsorted_op_test.py"], srcs = ["searchsorted_op_test.py"],
enable_mlir_bridge = True,
python_version = "PY3", python_version = "PY3",
tags = [ tags = [
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip "no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip
@ -691,6 +692,7 @@ tf_xla_py_test(
name = "fft_test", name = "fft_test",
size = "medium", size = "medium",
srcs = ["fft_test.py"], srcs = ["fft_test.py"],
enable_mlir_bridge = True,
python_version = "PY3", python_version = "PY3",
shard_count = 6, shard_count = 6,
tags = [ tags = [
@ -931,6 +933,7 @@ tf_xla_py_test(
name = "pooling_ops_test", name = "pooling_ops_test",
size = "medium", size = "medium",
srcs = ["pooling_ops_test.py"], srcs = ["pooling_ops_test.py"],
enable_mlir_bridge = True,
python_version = "PY3", python_version = "PY3",
shard_count = 20, shard_count = 20,
tags = [ tags = [
@ -1009,6 +1012,7 @@ tf_xla_py_test(
"cpu", "cpu",
"cpu_ondemand", "cpu_ondemand",
], ],
enable_mlir_bridge = True,
python_version = "PY3", python_version = "PY3",
shard_count = 5, shard_count = 5,
tags = [ tags = [
@ -1035,6 +1039,7 @@ tf_xla_py_test(
"cpu", "cpu",
"cpu_ondemand", "cpu_ondemand",
], ],
enable_mlir_bridge = True,
python_version = "PY3", python_version = "PY3",
shard_count = 5, shard_count = 5,
tags = [ tags = [
@ -1117,6 +1122,7 @@ tf_xla_py_test(
name = "reverse_ops_test", name = "reverse_ops_test",
size = "medium", size = "medium",
srcs = ["reverse_ops_test.py"], srcs = ["reverse_ops_test.py"],
enable_mlir_bridge = True,
python_version = "PY3", python_version = "PY3",
tags = [ tags = [
"no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip "no_pip", # TODO(b/149738646): fix pip install so these tests run on kokoro pip