From 6deb17ed8c5706835caffae94dcfa968d2151acb Mon Sep 17 00:00:00 2001
From: dzhwinter <dzhwinter@gmail.com>
Date: Tue, 19 Feb 2019 16:59:12 +0800
Subject: [PATCH] fix default value. test=develop

---
 paddle/fluid/framework/details/memory_optimize_pass.cc | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/framework/details/memory_optimize_pass.cc b/paddle/fluid/framework/details/memory_optimize_pass.cc
index d45a43d851d..fd02bc4697e 100644
--- a/paddle/fluid/framework/details/memory_optimize_pass.cc
+++ b/paddle/fluid/framework/details/memory_optimize_pass.cc
@@ -236,8 +236,12 @@ void MemoryOptimizePass::RenameVarInGraphDesc(const std::string& var,
     auto* op_desc = op->Op();
     op_desc->RenameInput(var, cache_var);
     op_desc->RenameOutput(var, cache_var);
-    if (op_desc->Block() != nullptr && op_desc->Block()->HasVar(var)) {
+    if (op_desc->Block() != nullptr) {
       op_desc->Block()->RemoveVar(var);
+    } else {
+      LOG(WARNING) << "op " << op->Name() << " not know its block."
+                   << "Is the op_desc created without block pointer? "
+                   << "Can not find " << var << " in Block(0)";
     }
     op_desc->Flush();
   }
-- 
GitLab