From 5474bf2eba0f6f5488763f791d236a3161ad629c Mon Sep 17 00:00:00 2001
From: chujinjin <chujinjin52@huawei.com>
Date: Mon, 29 Mar 2021 20:10:31 +0800
Subject: [PATCH] fix device memory not free for constexpr

---
 mindspore/ccsrc/pipeline/pynative/pynative_execute.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
index 86b46566c0..17947d256f 100644
--- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
+++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
@@ -812,8 +812,8 @@ abstract::AbstractBasePtr ForwardExecutor::CheckConstValue(const PrimitivePyPtr
       MS_EXCEPTION_IF_NULL(new_abs);
       new_abs = new_abs->Broaden(config);
       MS_LOG(DEBUG) << "Broaden for " << prim->ToString() << " " << config;
+      node_abs_map_[id] = new_abs;
     }
-    node_abs_map_[id] = new_abs;
   }
   return new_abs;
 }