From 64d8626d6d413bb2f4a266205acd538f506ab473 Mon Sep 17 00:00:00 2001 From: wanxing Date: Thu, 12 Dec 2019 13:49:52 +0800 Subject: [PATCH] Optimize ray::LocalMemoryBuffer performance (#6384) --- src/ray/common/buffer.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/ray/common/buffer.h b/src/ray/common/buffer.h index 5aba5c9d3..0311ea1f5 100644 --- a/src/ray/common/buffer.h +++ b/src/ray/common/buffer.h @@ -55,7 +55,8 @@ class LocalMemoryBuffer : public Buffer { : has_data_copy_(copy_data) { if (copy_data) { RAY_CHECK(data != nullptr); - buffer_.insert(buffer_.end(), data, data + size); + buffer_.resize(size); + std::copy(data, data + size, buffer_.begin()); data_ = buffer_.data(); size_ = buffer_.size(); } else { @@ -79,7 +80,7 @@ class LocalMemoryBuffer : public Buffer { bool IsPlasmaBuffer() const override { return false; } - ~LocalMemoryBuffer() {} + ~LocalMemoryBuffer() { size_ = 0; } private: /// Disable copy constructor and assignment, as default copy will