Method IncreasePoolSizeByBase

oneflow/core/common/tensor_buffer.cpp:271–276

}

void TensorBufferPool::IncreasePoolSizeByBase(size_t base) {
  std::unique_lock<std::mutex> lck(mtx_);
  pool_size_ += GetTensorBufferPoolSize(base);
  if (pool_size_ > global_free_list_.capacity()) { global_free_list_.reserve(pool_size_); }
  if (pool_size_ < global_free_list_.size()) { global_free_list_.resize(pool_size_); }
}

void TensorBufferPool::DecreasePoolSizeByBase(size_t base) {
  std::unique_lock<std::mutex> lck(mtx_);