未验证 提交 13f1050a 编写于 作者: D dzhwinter 提交者: GitHub

"fix mixed_vector bug" (#9319)

上级 3c8bbd30
...@@ -176,7 +176,7 @@ class Vector { ...@@ -176,7 +176,7 @@ class Vector {
// resize the vector // resize the vector
void resize(size_t size) { void resize(size_t size) {
if (size + 1 < capacity()) { if (size + 1 <= capacity()) {
size_ = size; size_ = size;
} else { } else {
MutableCPU(); MutableCPU();
......
...@@ -104,3 +104,11 @@ TEST(mixed_vector, ForEach) { ...@@ -104,3 +104,11 @@ TEST(mixed_vector, ForEach) {
for (auto& v : tmp) { for (auto& v : tmp) {
} }
} }
TEST(mixed_vector, Reserve) {
paddle::framework::Vector<int> vec;
vec.reserve(1);
vec.push_back(0);
vec.push_back(0);
vec.push_back(0);
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册