fix bug

7 years ago · 450c39a413
parent d5686f5831
commit 450c39a413
1 changed files with 1 additions and 0 deletions
--- 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);
 }