aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar Benjamin Kramer <kramerb@google.com>2018-09-19 12:03:27 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-19 12:08:00 -0700
commit238424ffcf04c38561ed48ebadb16b3b3a770e2e (patch)
treef0daea812d11b634fce71de52b273929a56b3f3d
parent43908cc8e3cba7fad329e72598107fe5bfff5a88 (diff)
[XLA:TF] Re-disable testRandomUniformIsInRange
The bug is still there and makes this test flakily fail with fp16. PiperOrigin-RevId: 213669453
-rw-r--r--tensorflow/compiler/tests/random_ops_test.py3
1 files changed, 2 insertions, 1 deletions
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():