update

src/framework/tensor.h

@@ -67,6 +67,15 @@ class Tensor : public TensorBase {
 
   /*! The internal of two tensors share the same memory block. */
   inline Tensor &ShareDataWith(const Tensor &src) {
+    src.check_memory_size();
+    if (holder_.get() != src.holder_.get()) {
+      *this = src;
+    }
+    return *this;
+  }
+
+  /*! The internal of two tensors share the same memory block. */
+  inline Tensor &ShareHolderWith(const Tensor &src) {
     src.check_memory_size();
     if (holder_.get() != src.holder_.get()) {
       holder_ = src.holder_;

src/pass/memory_optimize.cpp

@@ -131,7 +131,7 @@ void MemoryOptPass::operator()(const framework::ProgramDesc *program,
         DLOG << node->name;
         auto *var = scope->Var(node->name);
         auto *tensor = var->template GetMutable<framework::LoDTensor>();
-        tensor->ShareDataWith(*reuse_tensor);
+        tensor->ShareHolderWith(*reuse_tensor);
       }
     }
   }