Change reserve to resize

test=develop
release/1.1
minqiyang 6 years ago
parent 5660d6a3ba
commit 0de6811ee0

@ -114,8 +114,8 @@ class LookupTableGradKernel : public framework::OpKernel<T> {
int64_t ids_num = ids->numel();
std::vector<int64_t> new_rows;
new_rows.reserve(ids_num);
std::memcpy(new_rows.data(), ids_data, ids_num * sizeof(int64_t));
new_rows.resize(ids_num);
std::memcpy(&new_rows[0], ids_data, ids_num * sizeof(int64_t));
d_table->set_rows(new_rows);
auto *d_table_value = d_table->mutable_value();

Loading…
Cancel
Save