From 8a35913f70893c435ea40a16ab5eabc9c126ba23 Mon Sep 17 00:00:00 2001 From: Xinyi Wang Date: Fri, 9 Oct 2020 16:17:37 -0700 Subject: [PATCH] Disable mlir support for some tests that fail with it. PiperOrigin-RevId: 336385193 Change-Id: I75500fc14cfee384f43a8018e9c9e09654f171b9 --- tensorflow/python/keras/distribute/BUILD | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD index e1a01272918..f067cd1fdb1 100644 --- a/tensorflow/python/keras/distribute/BUILD +++ b/tensorflow/python/keras/distribute/BUILD @@ -335,6 +335,7 @@ distribute_py_test( distribute_py_test( name = "distribute_strategy_test", srcs = ["distribute_strategy_test.py"], + disable_mlir_bridge = True, # TODO(b/170352626) full_precision = True, main = "distribute_strategy_test.py", python_version = "PY3", @@ -402,7 +403,7 @@ distribute_py_test( name = "keras_dnn_correctness_test", size = "medium", srcs = ["keras_dnn_correctness_test.py"], - disable_mlir_bridge = False, + disable_mlir_bridge = True, # TODO(b/170352626) full_precision = True, main = "keras_dnn_correctness_test.py", # Shard count is set to an odd number to distribute tasks across @@ -423,6 +424,7 @@ distribute_py_test( name = "keras_embedding_model_correctness_test", size = "medium", srcs = ["keras_embedding_model_correctness_test.py"], + disable_mlir_bridge = True, # TODO(b/170352626) full_precision = True, main = "keras_embedding_model_correctness_test.py", shard_count = 8, @@ -440,7 +442,7 @@ distribute_py_test( name = "keras_image_model_correctness_test", size = "medium", srcs = ["keras_image_model_correctness_test.py"], - disable_mlir_bridge = False, + disable_mlir_bridge = True, # TODO(b/170352626) full_precision = True, main = "keras_image_model_correctness_test.py", shard_count = 16, @@ -553,6 +555,7 @@ distribute_py_test( distribute_py_test( name = "keras_utils_test", srcs = ["keras_utils_test.py"], + disable_mlir_bridge = True, # TODO(b/170352626) full_precision = True, main = "keras_utils_test.py", shard_count = 4,