提交 93779784 编写于 作者: Y Yi Wang

Fix serde_test.cc compile error

上级 73695060
...@@ -102,12 +102,12 @@ void RunSerdeTestSelectedRows(platform::Place place) { ...@@ -102,12 +102,12 @@ void RunSerdeTestSelectedRows(platform::Place place) {
} else { } else {
tensor_data2 = const_cast<float*>(tensor2->data<float>()); tensor_data2 = const_cast<float*>(tensor2->data<float>());
} }
const size_t* rows_data2 = rows2->data(); const int64_t* rows_data2 = rows2->data();
for (int i = 0; i < tensor_numel; ++i) { for (int i = 0; i < tensor_numel; ++i) {
EXPECT_FLOAT_EQ(tensor_data2[i], 32.7); EXPECT_FLOAT_EQ(tensor_data2[i], 32.7);
} }
for (size_t i = 0; i < rows2->size(); ++i) { for (int64_t i = 0; i < rows2->size(); ++i) {
EXPECT_EQ(rows_data2[i], i); EXPECT_EQ(rows_data2[i], i);
} }
EXPECT_EQ(slr2->height(), 1000); EXPECT_EQ(slr2->height(), 1000);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册