KenLM 5cc905bc2d214efa7de2db56a9a672b749a95591

1 files changed, 1 insertions, 3 deletions

diff --git a/klm/util/pool.cc b/klm/util/pool.cc
index db72a8ec..429ba158 100644
--- a/klm/util/pool.cc
+++ b/klm/util/pool.cc
@@ -25,9 +25,7 @@ void Pool::FreeAll() {
 }
 
 void *Pool::More(std::size_t size) {
-  // Double until we hit 2^21 (2 MB).  Then grow in 2 MB blocks. 
-  std::size_t desired_size = static_cast<size_t>(32) << std::min(static_cast<std::size_t>(16), free_list_.size());
-  std::size_t amount = std::max(desired_size, size);
+  std::size_t amount = std::max(static_cast<size_t>(32) << free_list_.size(), size);
   uint8_t *ret = static_cast<uint8_t*>(MallocOrThrow(amount));
   free_list_.push_back(ret);
   current_ = ret + size;