diff --git a/paddle/fluid/operators/detail/serde_test.cc b/paddle/fluid/operators/detail/serde_test.cc
index 6fb2369e6e40b9be4328f37075827cceaa68b284..f8cae6b26acf9d37ca286487065d70ede4c03120 100644
--- a/paddle/fluid/operators/detail/serde_test.cc
+++ b/paddle/fluid/operators/detail/serde_test.cc
@@ -102,12 +102,12 @@ void RunSerdeTestSelectedRows(platform::Place place) {
   } else {
     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) {
     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(slr2->height(), 1000);