diff --git a/paddle/fluid/distributed/service/communicator.cc b/paddle/fluid/distributed/service/communicator.cc index 6d26e6577473adaae5bdf2c7ca21f7c6401f0858..09f8db145a1a48bfc7e4cfdfbf66f0b49e26b6aa 100644 --- a/paddle/fluid/distributed/service/communicator.cc +++ b/paddle/fluid/distributed/service/communicator.cc @@ -290,7 +290,7 @@ void Communicator::RpcSendSparse(const std::string &var_name, int table_id, auto dim = tensor->value().dims()[1]; std::transform(tensor->rows().begin(), tensor->rows().end(), std::back_inserter(sparse_push_keys), - [&](int id) { return static_cast(id); }); + [&](int64_t id) { return static_cast(id); }); for (auto i = 0; i < static_cast(sparse_push_keys.size()); ++i) { push_g_vec.push_back(tensor->mutable_value()->data() + i * dim);