relax benchmark milliseconds threshold

Yangqing Jia · Yangqing Jia · commit 9ff22ec52831 · 2014-11-07T17:42:43.000-08:00
diff --git a/src/caffe/test/test_benchmark.cpp b/src/caffe/test/test_benchmark.cpp
@@ -9,6 +9,8 @@
 
 namespace caffe {
 
+const float kMillisecondsThreshold = 30;
+
 template <typename TypeParam>
 class BenchmarkTest : public MultiDeviceTest<TypeParam> {};
 
@@ -63,8 +65,8 @@ TYPED_TEST(BenchmarkTest, TestTimerMilliSeconds) {
   EXPECT_FALSE(timer.has_run_at_least_once());
   timer.Start();
   usleep(300 * 1000);
-  EXPECT_GE(timer.MilliSeconds(), 290);
-  EXPECT_LE(timer.MilliSeconds(), 310);
+  EXPECT_GE(timer.MilliSeconds(), 300 - kMillisecondsThreshold);
+  EXPECT_LE(timer.MilliSeconds(), 300 + kMillisecondsThreshold);
   EXPECT_TRUE(timer.initted());
   EXPECT_FALSE(timer.running());
   EXPECT_TRUE(timer.has_run_at_least_once());
@@ -78,8 +80,8 @@ TYPED_TEST(BenchmarkTest, TestTimerSeconds) {
   EXPECT_FALSE(timer.has_run_at_least_once());
   timer.Start();
   usleep(300 * 1000);
-  EXPECT_GE(timer.Seconds(), 0.290);
-  EXPECT_LE(timer.Seconds(), 0.310);
+  EXPECT_GE(timer.Seconds(), 0.3 - kMillisecondsThreshold / 1000.);
+  EXPECT_LE(timer.Seconds(), 0.3 + kMillisecondsThreshold / 1000.);
   EXPECT_TRUE(timer.initted());
   EXPECT_FALSE(timer.running());
   EXPECT_TRUE(timer.has_run_at_least_once());
