fix reference_count_pass,test=develop (#17060)

test=develop
6 years ago · 842ded14b0
parent beda78258f
commit 842ded14b0
1 changed files with 1 additions and 0 deletions
--- a/paddle/fluid/framework/details/reference_count_pass.cc
+++ b/paddle/fluid/framework/details/reference_count_pass.cc
@ -138,6 +138,7 @@ static ComputationOpHandle *FindNextComputationOpHandleOrReturnItself(
      for (auto *pending_op : out_var->PendingOps()) {
        if (visited.count(pending_op)) continue;
        visited.insert(pending_op);
+        q.push(pending_op);
      }
    }
  } while (!q.empty());