fix bug

450c39a4 · Kexin Zhao · d5686f58 · 450c39a4
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

paddle/framework/prune.cc paddle/framework/prune.cc +1 -0

未找到文件。
--- a/paddle/framework/prune.cc
+++ b/paddle/framework/prune.cc
@@ -182,6 +182,7 @@ void prune_impl(const proto::ProgramDesc& input, proto::ProgramDesc* output,
 // TODO(fengjiayi): Prune() could be inplaced to avoid unnecessary copies
 void Prune(const proto::ProgramDesc& input, proto::ProgramDesc* output) {
  std::set<std::string> dependent_vars;
+  output->clear_blocks();
  prune_impl(input, output, 0, -1, dependent_vars);
 }