From 238424ffcf04c38561ed48ebadb16b3b3a770e2e Mon Sep 17 00:00:00 2001 From: Benjamin Kramer Date: Wed, 19 Sep 2018 12:03:27 -0700 Subject: [XLA:TF] Re-disable testRandomUniformIsInRange The bug is still there and makes this test flakily fail with fp16. PiperOrigin-RevId: 213669453 --- tensorflow/compiler/tests/random_ops_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/compiler/tests/random_ops_test.py b/tensorflow/compiler/tests/random_ops_test.py index c423fa5004..36ef6ed5fe 100644 --- a/tensorflow/compiler/tests/random_ops_test.py +++ b/tensorflow/compiler/tests/random_ops_test.py @@ -76,7 +76,8 @@ class RandomOpsTest(xla_test.XLATestCase): for dtype in self._random_types(): # TODO (b/112272078): enable bfloat16 for CPU and GPU when the bug is # fixed. - if (self.device in ["XLA_GPU", "XLA_CPU"]) and (dtype == dtypes.bfloat16): + if (self.device in ["XLA_GPU", "XLA_CPU" + ]) and (dtype in [dtypes.bfloat16, dtypes.half]): continue with self.cached_session() as sess: with self.test_scope(): -- cgit v1.2.3