From 44597e39fb9e6eeb614d2119d516c0d4ede084cc Mon Sep 17 00:00:00 2001 From: Yunxing Dai Date: Sat, 1 Aug 2020 02:40:22 -0700 Subject: [PATCH] Enforce xla in xla_control_flow_ops_test. I actually don't understand how one can avoid using xla in an xla test. But the test seems to pass now. PiperOrigin-RevId: 324379840 Change-Id: I62c8638183148a95c9c5ab930fa64cfb4c66926c --- .../python/ops/parallel_for/xla_control_flow_ops_test.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tensorflow/python/ops/parallel_for/xla_control_flow_ops_test.py b/tensorflow/python/ops/parallel_for/xla_control_flow_ops_test.py index b1762e2f55f..33f0d7b76ae 100644 --- a/tensorflow/python/ops/parallel_for/xla_control_flow_ops_test.py +++ b/tensorflow/python/ops/parallel_for/xla_control_flow_ops_test.py @@ -233,8 +233,7 @@ class WhileV2Test(PForTestCase): body, [True, 0, 0.]) - # b/155430349: Enabling forrce_xla=True triggers a CHECK in debug mode. - self._test_loop_fn(loop_fn, 3, force_xla=False) + self._test_loop_fn(loop_fn, 3, force_xla=True) if __name__ == "__main__":