fix default value. test=develop

6 years ago · 9c92d0304f
parent a661d0bd1d
commit 9c92d0304f
1 changed files with 3 additions and 1 deletions
--- a/paddle/fluid/framework/details/memory_optimize_pass.cc
+++ b/paddle/fluid/framework/details/memory_optimize_pass.cc
@ -235,7 +235,9 @@ 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()->HasVar(var)) op_desc->Block()->RemoveVar(var);
+    if (op_desc->Block() != nullptr && op_desc->Block()->HasVar(var)) {
+      op_desc->Block()->RemoveVar(var);
+    }
    op_desc->Flush();
  }
 }