修改第二课

zjhellofss · Jun 22, 2024 · 58ca115 · 58ca115
1 parent e43d937
commit 58ca115
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/kuiper/source/base/buffer.cpp b/kuiper/source/base/buffer.cpp
@@ -18,6 +18,7 @@ Buffer::Buffer(size_t byte_size, std::shared_ptr<DeviceAllocator> allocator, voi
 Buffer::~Buffer() {
   if (!use_external_) {
     if (ptr_ && allocator_) {
+      LOG(INFO) << "Release...";
       allocator_->release(ptr_);
       ptr_ = nullptr;
     }

diff --git a/test/test_buffer.cpp b/test/test_buffer.cpp
@@ -2,11 +2,15 @@
 #include <gtest/gtest.h>
 #include "base/buffer.h"
 
-TEST(test_buffer, allocate) {
+
+
+TEST(test_buffer, allocate2) {
   using namespace base;
   auto alloc = base::CPUDeviceAllocatorFactory::get_instance();
-  Buffer buffer(32, alloc);
-  ASSERT_NE(buffer.ptr(), nullptr);
+  std::shared_ptr<Buffer> buffer;
+  { buffer = std::make_shared<Buffer>(32, alloc); }
+  LOG(INFO) << "HERE";
+  ASSERT_NE(buffer->ptr(), nullptr);
 }
 
 TEST(test_buffer, use_external) {