Skip to content

Commit 0de6811

Browse files
committed
Change reserve to resize
test=develop
1 parent 5660d6a commit 0de6811

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/fluid/operators/lookup_table_op.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,8 +114,8 @@ class LookupTableGradKernel : public framework::OpKernel<T> {
114114
int64_t ids_num = ids->numel();
115115

116116
std::vector<int64_t> new_rows;
117-
new_rows.reserve(ids_num);
118-
std::memcpy(new_rows.data(), ids_data, ids_num * sizeof(int64_t));
117+
new_rows.resize(ids_num);
118+
std::memcpy(&new_rows[0], ids_data, ids_num * sizeof(int64_t));
119119
d_table->set_rows(new_rows);
120120

121121
auto *d_table_value = d_table->mutable_value();

0 commit comments

Comments
 (0)