From 222159599a70c647bd9bd06ba503f1497b20681a Mon Sep 17 00:00:00 2001
From: laiyongqiang <laiyongqiang1@huawei.com>
Date: Sat, 14 Nov 2020 15:23:14 +0800
Subject: [PATCH] independent node use the somas's memory

---
 mindspore/ccsrc/backend/optimizer/somas/somas.cc | 2 +-
 mindspore/ccsrc/runtime/device/kernel_runtime.cc | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/mindspore/ccsrc/backend/optimizer/somas/somas.cc b/mindspore/ccsrc/backend/optimizer/somas/somas.cc
index 030c3a08a5..c4a6ff3427 100644
--- a/mindspore/ccsrc/backend/optimizer/somas/somas.cc
+++ b/mindspore/ccsrc/backend/optimizer/somas/somas.cc
@@ -301,7 +301,7 @@ void Somas::IndependentNodeOutputProcess(const session::KernelGraph *graph) {
       auto semi_reuse_output_tensors = iter->second->output_tensors_;
       for (auto &tensor : semi_reuse_output_tensors) {
         total_size += tensor->GetAlignedSize();
-        tensor->aligned_size_ = 0;
+        tensor->lifelong_value_ = kLifeLongGraphAll;
       }
     }
   }
diff --git a/mindspore/ccsrc/runtime/device/kernel_runtime.cc b/mindspore/ccsrc/runtime/device/kernel_runtime.cc
index 5732df7f91..8b2cf4904f 100644
--- a/mindspore/ccsrc/runtime/device/kernel_runtime.cc
+++ b/mindspore/ccsrc/runtime/device/kernel_runtime.cc
@@ -543,7 +543,7 @@ void KernelRuntime::AssignNodeOutputMem(MemType type, const AnfNodePtr &node, in
 
   if (node->isa<CNode>()) {
     bool independent = AnfAlgo::IsIndependentNode(node->cast<CNodePtr>());
-    if (independent && (type == kReuseDynamicMem || type == kSomasReuseDynamicMem)) {
+    if (independent && (type == kReuseDynamicMem)) {
       MS_LOG(INFO) << "Independent node " << node->fullname_with_scope() << " disable memory reuse";
       type = kDynamicMem;
     }