From 1e3981b63885401e9a308d02d6468eec7bad2d4d Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 13 Feb 2019 13:23:27 -0800 Subject: [PATCH] [TF:XLA] Enable some tests that were mistakenly disabled. PiperOrigin-RevId: 233816788 --- tensorflow/python/BUILD | 1 + tensorflow/python/ops/bitwise_ops_test.py | 3 --- tensorflow/python/ops/special_math_ops_test.py | 1 - 3 files changed, 1 insertion(+), 4 deletions(-) diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 542a8fd55e8..79f4c68ac4f 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -3403,6 +3403,7 @@ cuda_py_test( ":framework_test_lib", ], tags = ["no_windows"], + xla_enable_strict_auto_jit = True, ) cuda_py_test( diff --git a/tensorflow/python/ops/bitwise_ops_test.py b/tensorflow/python/ops/bitwise_ops_test.py index c182874c7f2..d154b6759bf 100644 --- a/tensorflow/python/ops/bitwise_ops_test.py +++ b/tensorflow/python/ops/bitwise_ops_test.py @@ -35,7 +35,6 @@ class BitwiseOpTest(test_util.TensorFlowTestCase): super(BitwiseOpTest, self).__init__(method_name) @test_util.run_deprecated_v1 - @test_util.disable_xla("This test never passed for XLA") def testBinaryOps(self): dtype_list = [dtypes.int8, dtypes.int16, dtypes.int32, dtypes.int64, dtypes.uint8, dtypes.uint16, dtypes.uint32, dtypes.uint64] @@ -73,7 +72,6 @@ class BitwiseOpTest(test_util.TensorFlowTestCase): self.assertAllEqual(truth, popcnt_result) @test_util.run_deprecated_v1 - @test_util.disable_xla("This test never passed for XLA") def testInvertOp(self): dtype_list = [dtypes.int8, dtypes.int16, dtypes.int32, dtypes.int64, dtypes.uint8, dtypes.uint16, dtypes.uint32, dtypes.uint64] @@ -99,7 +97,6 @@ class BitwiseOpTest(test_util.TensorFlowTestCase): self.assertAllEqual(inverted, expected) @test_util.run_deprecated_v1 - @test_util.disable_xla("This test never passed for XLA") def testShiftsWithPositiveLHS(self): dtype_list = [np.int8, np.int16, np.int32, np.int64, np.uint8, np.uint16, np.uint32, np.uint64] diff --git a/tensorflow/python/ops/special_math_ops_test.py b/tensorflow/python/ops/special_math_ops_test.py index 60e0f1f832b..41ba060a4b6 100644 --- a/tensorflow/python/ops/special_math_ops_test.py +++ b/tensorflow/python/ops/special_math_ops_test.py @@ -119,7 +119,6 @@ class LBetaTest(test.TestCase): special_math_ops.lbeta(x).get_shape()) @test_util.run_in_graph_and_eager_modes - @test_util.disable_xla('This test never passed for XLA') def test_length_1_last_dimension_results_in_one(self): # If there is only one coefficient, the formula still works, and we get one # as the answer, always.