fix batch merge bug (#16601)

6 years ago · 423bc515da
parent 0342f01249
commit 423bc515da
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/framework/ir/multi_batch_merge_pass.cc
+++ b/paddle/fluid/framework/ir/multi_batch_merge_pass.cc
@ -84,7 +84,8 @@ void BatchMergePass::ApplyImpl(ir::Graph* graph) const {

  // 1. record op nodes of different roles
  for (auto node : nodes) {
-    if (node->IsVar()) continue;
+    if (!node->IsOp()) continue;
+    PADDLE_ENFORCE(node->Op(), "must find opdesc");
    int op_role = boost::get<int>(node->Op()->GetAttr(
        framework::OpProtoAndCheckerMaker::OpRoleAttrName()));
    if ((op_role == static_cast<int>(framework::OpRole::kForward)) ||