fix ShmSetUp error

Signed-off-by: daquexian <[email protected]>
Oneflow-Inc · Mar 8, 2023 · 0279c6e · 0279c6e
1 parent 4cf27fa
commit 0279c6e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/oneflow/core/eager/tensor_storage.cpp b/oneflow/core/eager/tensor_storage.cpp
@@ -27,8 +27,11 @@ TensorStorage::TensorStorage(bool is_allocated_in_vm)
       last_used_stream_(NullOpt),
       is_allocated_in_vm_(is_allocated_in_vm) {}
 
-void TensorStorage::Release() {
+TensorStorage::~TensorStorage() {
   for (const auto& hook : storage_delete_hooks_) { hook(); }
+}
+
+void TensorStorage::Release() {
   non_pod_allocator_.reset();
   blob_dptr_.reset();
 }

diff --git a/oneflow/core/eager/tensor_storage.h b/oneflow/core/eager/tensor_storage.h
@@ -30,7 +30,7 @@ class TensorStorage {
   explicit TensorStorage(bool is_allocated_in_vm);
   OF_DISALLOW_COPY_AND_MOVE(TensorStorage);
 
-  virtual ~TensorStorage() = default;
+  virtual ~TensorStorage();
 
   bool is_allocated_in_vm() const { return is_allocated_in_vm_; }