diff options
-rw-r--r-- | tensorflow/core/debug/debug_gateway_test.cc | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/tensorflow/core/debug/debug_gateway_test.cc b/tensorflow/core/debug/debug_gateway_test.cc index 963cea8419..d6f656c5ca 100644 --- a/tensorflow/core/debug/debug_gateway_test.cc +++ b/tensorflow/core/debug/debug_gateway_test.cc @@ -336,6 +336,10 @@ TEST_F(SessionDebugMinusAXTest, RunSimpleNetworkWithTwoDebugNodesInserted) { ASSERT_EQ(1, debug_nan_count_tensor_vals[0].scalar<int64>()()); } +#ifndef GOOGLE_CUDA +// TODO(cais): Reinstate the following test for concurrent debugged runs on +// a GPU once the root cause of the ~0.5% flakiness has been addressed. +// (b/34081273) TEST_F(SessionDebugMinusAXTest, RunSimpleNetworkConcurrentlyWithDifferentDebugTensorWatches) { // Test concurrent Run() calls on a graph with different debug watches. @@ -442,11 +446,7 @@ TEST_F(SessionDebugMinusAXTest, &outputs, &run_metadata); TF_ASSERT_OK(s); -#if GOOGLE_CUDA - ASSERT_EQ(2, run_metadata.partition_graphs().size()); -#else ASSERT_EQ(1, run_metadata.partition_graphs().size()); -#endif ASSERT_EQ(1, outputs.size()); ASSERT_TRUE(outputs[0].IsInitialized()); @@ -490,6 +490,7 @@ TEST_F(SessionDebugMinusAXTest, ASSERT_EQ(-1.0, y_mat_identity(1, 0)); } } +#endif class SessionDebugOutputSlotWithoutOngoingEdgeTest : public ::testing::Test { public: |