From bb0ffa51024d5f3c140e6e506ceca9fb2a238a7a Mon Sep 17 00:00:00 2001
From: Pieter Ghysels <pghysels@lbl.gov>
Date: Thu, 11 Jul 2024 12:42:51 -0700
Subject: [PATCH] Clean up the vectorpool, no more than 4 vectors

---
 src/misc/Tools.hpp | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/misc/Tools.hpp b/src/misc/Tools.hpp
index 255a8b2d..ec04410a 100644
--- a/src/misc/Tools.hpp
+++ b/src/misc/Tools.hpp
@@ -109,7 +109,22 @@ namespace strumpack {
     void restore(std::vector<scalar_t,NoInit<scalar_t>>& v) {
       if (v.empty()) return;
 #pragma omp critical
-      data_.push_back(std::move(v));
+      {
+        data_.push_back(std::move(v));
+        if (data_.size() > 4) {
+          // remove smallest
+          int pos = 0;
+          std::size_t smin = data_[0].size();
+          for (std::size_t i=1; i<data_.size(); i++) {
+            auto ps = data_[i].size();
+            if (ps < smin) {
+              pos = i;
+              smin = ps;
+            }
+          }
+          data_.erase(data_.begin()+pos);
+        }
+      }
     }
 
 #if defined(STRUMPACK_USE_GPU)