| 276 | } |
| 277 | |
| 278 | void TensorBufferPool::DecreasePoolSizeByBase(size_t base) { |
| 279 | std::unique_lock<std::mutex> lck(mtx_); |
| 280 | size_t dec = GetTensorBufferPoolSize(base); |
| 281 | CHECK_GE(pool_size_, dec) << "pool_size " << pool_size_ << " decreased by " << dec |
| 282 | << " would be negative"; |
| 283 | pool_size_ -= dec; |
| 284 | if (pool_size_ > global_free_list_.capacity()) { global_free_list_.reserve(pool_size_); } |
| 285 | if (pool_size_ < global_free_list_.size()) { global_free_list_.resize(pool_size_); } |
| 286 | } |
| 287 | |
| 288 | } // namespace oneflow |
no test coverage detected