Skip to content

Commit 5a748a7

Browse files
committed
Merge pull request #1 from ibmsoe/fix-EmbedLayerTest
Fix for Bug# 140953 - https://bugzilla.linux.ibm.com/show_bug.cgi?id=…
2 parents e88cbe5 + 4e18897 commit 5a748a7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/caffe/test/test_embed_layer.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ TYPED_TEST(EmbedLayerTest, TestForwardWithBias) {
124124
top_offset[4] = 0;
125125
bias_offset[0] = 0;
126126
for (int j = 0; j < kNumOutput; ++j) {
127-
EXPECT_EQ(layer->blobs()[0]->data_at(weight_offset) +
127+
EXPECT_FLOAT_EQ(layer->blobs()[0]->data_at(weight_offset) +
128128
layer->blobs()[1]->data_at(bias_offset),
129129
this->blob_top_->data_at(top_offset));
130130
++top_offset[4];

0 commit comments

Comments
 (0)