From: A. Unique TensorFlower <gardener@tensorflow.org>
Date: Fri, 27 Apr 2018 22:28:01 +0000 (-0700)
Subject: Fix broken ElementWiseFusionTest.
X-Git-Tag: upstream/v1.9.0_rc1~190^2^2~22
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=fbd9ecd7361ff384bc05e30d2b44fc2a1f1da72b;p=platform%2Fupstream%2Ftensorflow.git

Fix broken ElementWiseFusionTest.

PiperOrigin-RevId: 194602336
---

diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD
index 991e65c..6a7b8fa 100644
--- a/tensorflow/compiler/tests/BUILD
+++ b/tensorflow/compiler/tests/BUILD
@@ -821,11 +821,6 @@ cuda_py_test(
         "//tensorflow/python:math_ops",
         "//tensorflow/python:nn_ops",
     ],
-    # TODO(b/62961789): Test fails with SIGABRT
-    tags = [
-        "manual",
-        "notap",
-    ],
 )
 
 cc_library(
diff --git a/tensorflow/compiler/tests/jit_test.py b/tensorflow/compiler/tests/jit_test.py
index 1f7da65..0310cdd 100644
--- a/tensorflow/compiler/tests/jit_test.py
+++ b/tensorflow/compiler/tests/jit_test.py
@@ -489,7 +489,8 @@ class ElementWiseFusionTest(test.TestCase):
   def testElementWiseClustering(self):
     arg0 = np.random.rand(2, 2).astype(np.float32)
     arg1 = np.random.rand(2, 2).astype(np.float32)
-    os.environ["TF_XLA_FLAGS"] = "--tf_xla_fusion_only=true"
+    os.environ["TF_XLA_FLAGS"] = ("--tf_xla_fusion_only=true "
+                                  "--tf_xla_cpu_global_jit")
     tf_op, tf_count = self.simpleTest(arg0, arg1,
                                       config_pb2.OptimizerOptions.OFF)
     self.assertEqual(0, tf_count)