[fix] fix plain_device mem leak

OpenPPL · May 24, 2024 · 6631b8f · 6631b8f
1 parent 2cfd14c
commit 6631b8f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/ppl/nn/engines/llm_cuda/plain_device.cc b/src/ppl/nn/engines/llm_cuda/plain_device.cc
@@ -35,6 +35,10 @@ RetCode PlainDevice::DoInit() {
     return RC_SUCCESS;
 }
 
+PlainDevice::~PlainDevice() {
+    DoDestroy();
+}
+
 void PlainDevice::DoDestroy() {
     delete allocator_;
 }

diff --git a/src/ppl/nn/engines/llm_cuda/plain_device.h b/src/ppl/nn/engines/llm_cuda/plain_device.h
@@ -27,6 +27,7 @@ namespace ppl { namespace nn { namespace llm { namespace cuda {
 class PlainDevice final : public LlmCudaDevice {
 public:
     PlainDevice(bool async) : async_(async) {}
+    ~PlainDevice();
     using LlmCudaDevice::Realloc;
     ppl::common::RetCode Realloc(uint64_t bytes, BufferDesc*) override;
     void Free(BufferDesc*) override;