fix int64 bug (#30780)

fix push sparse int64 bug
revert-31068-fix_conv3d_windows
Chengmo 5 years ago committed by GitHub
parent 69875dc42c
commit d3fac0ea85
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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<uint64_t>(id); });
[&](int64_t id) { return static_cast<uint64_t>(id); });
for (auto i = 0; i < static_cast<int>(sparse_push_keys.size()); ++i) {
push_g_vec.push_back(tensor->mutable_value()->data<float>() + i * dim);

Loading…
Cancel
Save