Fix for a random failure in this test due to floating point compariso…

…n. So, instead of exact match, used EXPECT_FLOAT_EQ that tolerates some precision while comparing two floats
BVLC · Jul 11, 2016 · f1a8470 · f1a8470 · cdluminate · Jul 28, 2016
1 parent 6f5f7c5
commit f1a8470
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/caffe/test/test_embed_layer.cpp b/src/caffe/test/test_embed_layer.cpp
@@ -124,7 +124,7 @@ TYPED_TEST(EmbedLayerTest, TestForwardWithBias) {
     top_offset[4] = 0;
     bias_offset[0] = 0;
     for (int j = 0; j < kNumOutput; ++j) {
-      EXPECT_EQ(layer->blobs()[0]->data_at(weight_offset) +
+      EXPECT_FLOAT_EQ(layer->blobs()[0]->data_at(weight_offset) +
                 layer->blobs()[1]->data_at(bias_offset),
                 this->blob_top_->data_at(top_offset));
       ++top_offset[4];