fix default value. test=develop

9c92d030 · dzhwinter · a661d0bd · 9c92d030
显示空白变更内容
内联并排

Showing with 3 addition and 1 deletion

paddle/fluid/framework/details/memory_optimize_pass.cc paddle/fluid/framework/details/memory_optimize_pass.cc +3 -1

未找到文件。
--- 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();
  }
 }