Skip to content

Commit 13f1050

Browse files
authored
"fix mixed_vector bug" (#9319)
1 parent 3c8bbd3 commit 13f1050

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

paddle/fluid/framework/mixed_vector.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ class Vector {
176176

177177
// resize the vector
178178
void resize(size_t size) {
179-
if (size + 1 < capacity()) {
179+
if (size + 1 <= capacity()) {
180180
size_ = size;
181181
} else {
182182
MutableCPU();

paddle/fluid/framework/mixed_vector_test.cu

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,3 +104,11 @@ TEST(mixed_vector, ForEach) {
104104
for (auto& v : tmp) {
105105
}
106106
}
107+
108+
TEST(mixed_vector, Reserve) {
109+
paddle::framework::Vector<int> vec;
110+
vec.reserve(1);
111+
vec.push_back(0);
112+
vec.push_back(0);
113+
vec.push_back(0);
114+
}

0 commit comments

Comments
 (0)