diff options
Diffstat (limited to 'src/residual_buffer_pool.cc')
| -rw-r--r-- | src/residual_buffer_pool.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/residual_buffer_pool.cc b/src/residual_buffer_pool.cc index e166392..44a842c 100644 --- a/src/residual_buffer_pool.cc +++ b/src/residual_buffer_pool.cc @@ -129,7 +129,8 @@ std::unique_ptr<ResidualBuffer> ResidualBufferPool::Get() { } void ResidualBufferPool::Release(std::unique_ptr<ResidualBuffer> buffer) { - buffer->transform_parameters()->Reset(); + buffer->transform_parameters()->Clear(); + buffer->partition_tree_order()->Clear(); std::lock_guard<std::mutex> lock(mutex_); buffers_.Push(std::move(buffer)); } |