From ad27775640e0d9c506247e91a2306c63703c97d0 Mon Sep 17 00:00:00 2001 From: Saurabh Saxena Date: Thu, 13 Jun 2019 01:16:02 -0700 Subject: [PATCH] Enable critical_section_test with control flow v2. PiperOrigin-RevId: 252982317 --- tensorflow/python/kernel_tests/critical_section_test.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tensorflow/python/kernel_tests/critical_section_test.py b/tensorflow/python/kernel_tests/critical_section_test.py index cc719d646a6..b233fb36637 100644 --- a/tensorflow/python/kernel_tests/critical_section_test.py +++ b/tensorflow/python/kernel_tests/critical_section_test.py @@ -34,6 +34,7 @@ from tensorflow.python.platform import tf_logging as logging # from tensorflow.python.training import saver as saver_lib +@test_util.with_control_flow_v2 class CriticalSectionTest(test.TestCase): @test_util.run_in_graph_and_eager_modes @@ -55,6 +56,8 @@ class CriticalSectionTest(test.TestCase): self.assertAllClose([2.0 * i for i in range(num_concurrent)], sorted(r_value)) + @test_util.disable_control_flow_v2("b/135070612") + @test_util.run_v1_only("b/135070612") @test_util.run_in_graph_and_eager_modes @test_util.xla_allow_fallback("b/128495870") def testCriticalSectionWithControlFlow(self):