From bcefec3d6782365510c45e08763892d478dabb07 Mon Sep 17 00:00:00 2001 From: Shashi Shekhar Date: Thu, 26 Apr 2018 16:11:11 -0700 Subject: [PATCH] Fix some flakiness in test. PiperOrigin-RevId: 194470125 --- tensorflow/contrib/lite/profiling/profiler_test.cc | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/tensorflow/contrib/lite/profiling/profiler_test.cc b/tensorflow/contrib/lite/profiling/profiler_test.cc index 7914f36..7ea1d8f 100644 --- a/tensorflow/contrib/lite/profiling/profiler_test.cc +++ b/tensorflow/contrib/lite/profiling/profiler_test.cc @@ -82,16 +82,15 @@ TEST(ProfilingTest, ProfilesAreCollected) { EXPECT_EQ("Child", profile_events[3]->tag); EXPECT_EQ("SleepForQuarter", profile_events[4]->tag); - AssertDurationOfEventAroundMs(profile_events[0], /*expected_ms*/ 500, - /*eps_ms*/ 2); - AssertDurationOfEventAroundMs(profile_events[1], /*expected_ms*/ 250, - /*eps_ms*/ 2); - AssertDurationOfEventAroundMs(profile_events[2], /*expected_ms*/ 250, - /*eps_ms*/ 2); - AssertDurationOfEventAroundMs(profile_events[3], /*expected_ms*/ 250, - /*eps_ms*/ 2); - AssertDurationOfEventAroundMs(profile_events[4], /*expected_ms*/ 250, - /*eps_ms*/ 2); +#ifndef ADDRESS_SANITIZER + // ASAN build is sometimes very slow. + const int eps_ms = 10; + AssertDurationOfEventAroundMs(profile_events[0], /*expected_ms*/ 500, eps_ms); + AssertDurationOfEventAroundMs(profile_events[1], /*expected_ms*/ 250, eps_ms); + AssertDurationOfEventAroundMs(profile_events[2], /*expected_ms*/ 250, eps_ms); + AssertDurationOfEventAroundMs(profile_events[3], /*expected_ms*/ 250, eps_ms); + AssertDurationOfEventAroundMs(profile_events[4], /*expected_ms*/ 250, eps_ms); +#endif } } // namespace -- 2.7.4