reserve storage in matrix_data constructor

ginkgo-project · May 9, 2022 · a4174ec · a4174ec
1 parent cbb53cf
commit a4174ec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/include/ginkgo/core/base/matrix_data.hpp b/include/ginkgo/core/base/matrix_data.hpp
@@ -169,6 +169,7 @@ struct matrix_data {
         if (is_zero(value)) {
             return;
         }
+        nonzeros.reserve(size[0] * size[1]);
         for (size_type row = 0; row < size[0]; ++row) {
             for (size_type col = 0; col < size[1]; ++col) {
                 nonzeros.emplace_back(row, col, value);
@@ -190,6 +191,7 @@ struct matrix_data {
     matrix_data(dim<2> size_, RandomDistribution&& dist, RandomEngine&& engine)
         : size{size_}
     {
+        nonzeros.reserve(size[0] * size[1]);
         for (size_type row = 0; row < size[0]; ++row) {
             for (size_type col = 0; col < size[1]; ++col) {
                 const auto value =